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

Code and data for Katsumi et al., in press, "Correspondence of functional connectivity gradients across human isocortex, cerebellum, and hippocampus".

Here is a Google Colab Notebook for fine-tuning Alpaca Lora (within 3 hours with a 40GB A100 GPU)