Stars
Implement a ChatGPT-like LLM in PyTorch from scratch, step by step
PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks
verl: Volcano Engine Reinforcement Learning for LLMs
[NeurIPS 2025] Reinforcement Learning for Reasoning in Large Language Models with One Training Example
PyTorch implementation of the groundbreaking paper "NoProp: Training Neural Networks Without Backpropagation or Forward Propagation".
GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projection
Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"
Hugging Face examples on NERSC systems
The WeightWatcher tool for predicting the accuracy of Deep Neural Networks
A feature-rich command-line audio/video downloader
a simple unofficial implementation of classifier-free diffusion guidance
Release for Improved Denoising Diffusion Probabilistic Models
Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!
Open-Sora: Democratizing Efficient Video Production for All
Course to get into Large Language Models (LLMs) with roadmaps and Colab notebooks.
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
Pruning is all you need (hopefully)
A method to increase the speed and lower the memory footprint of existing vision transformers.
Pytorch implementation of Diffusion Models (https://arxiv.org/pdf/2006.11239.pdf)
Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
An implementation of "Retentive Network: A Successor to Transformer for Large Language Models"