AI agents running research on single-GPU nanochat training automatically

This repository contains a curated collection of 300+ case studies from over 80 companies, detailing practical applications and insights into machine learning (ML) system design. The contents are o…

Proximal Policy Optimization (Continuous Version) in PyTorch.

Code for the paper "Evolution Strategies as a Scalable Alternative to Reinforcement Learning"

Seamless operability between C++11 and Python

[NeurIPS 2025] Latent Zoning Networks

Nano vLLM

A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.

code related to the OpenFMRI project

360-node volumetric atlas derived from the multimodal parcellation from Glasser and colleagues

Python implementation of Robust Continuous Clustering

3.8% and 18.3% on CIFAR-10 and CIFAR-100

simple Differential Privacy in PyTorch

Data Shapley: Equitable Valuation of Data for Machine Learning

Efficient Computation and Analysis of Distributional Shapley Values (AISTATS 2021)

Distributional Shapley: A Distributional Framework for Data Valuation

Wide Residual Networks (WideResNets) in PyTorch

Code for Auditing DPSGD

Code used for "Gradient flow dynamics of shallow ReLU networks for square loss and orthogonal inputs"

Official codebase for LEAP: Planning with Goal Conditioned Policies

1000 images, one per image-net class. For easy visualization/exploration of classes.

Beta Shapley: a Unified and Noise-reduced Data Valuation Framework for Machine Learning (AISTATS 2022 Oral)

Learning to See by Looking at Noise

A modified version of MineDojo