Code for our TMLR '24-ICLR '25 work: MMD-Regularized UOT.

This is the GitHub repository for STELLOPT, the state-of-the-art stellarator optimization code.

Differentiable neuron simulations with biophysical detail on CPU, GPU, or TPU.

A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.

BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.

This is a workshop designed for Amazon Bedrock a foundational model service.

LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.

Personal Webpage

My curriculum vitae (CV) written using LaTeX.

This resume can be used as LaTeX template for academic purposes or job applications.

FAST Randomized SVD on a GPU with CUDA 🏎️

CUDA Library Samples

Notebooks using the Hugging Face libraries 🤗

📄 Awesome CV is LaTeX template for your outstanding job application

The hub for EleutherAI's work on interpretability and learning dynamics

This repository contains examples for customers to get started using the Amazon Bedrock Service. This contains examples for all available foundational models

A collection of notebooks/recipes showcasing some fun and effective ways of using Claude.

Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)

MetaDrive: Lightweight driving simulator for everyone

A large-scale benchmark and learning environment.

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

Simplifying reinforcement learning for complex game environments

LaTeX style file for the Journal of Machine Learning Research

Simple and readable code for training and sampling from diffusion models

LLM inference in C/C++

The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”