JAX in JavaScript – ML library for the web, running on WebGPU & Wasm

FlashInfer: Kernel Library for LLM Serving

slime is an LLM post-training framework for RL Scaling.

Minimal and annotated implementations of key ideas from modern deep learning research.

Nano vLLM

Solve puzzles. Learn CUDA.

Meta Lingua: a lean, efficient, and easy-to-hack codebase to research LLMs.

Stanford NLP Python library for Representation Finetuning (ReFT)

Stanford NLP Python library for understanding and improving PyTorch models via interventions

Official PyTorch Implementation of "Rosetta Neurons: Mining the Common Units in a Model Zoo"