Soumith Chintala

@soumith

Author of PyTorch

View on GitHub

Authored Projects (1)

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Lilian Weng(Cofounder of Thinking Machines Lab), and

8 more.

ganhacks by soumith

0.1%

12k

GAN training tips and tricks

created 8 years ago

updated 3 years ago

Starred Projects (87)

Starred by

Tobi Lutke(Cofounder of Shopify),

Didier Lopes(Founder of OpenBB), and

7 more.

opencode by opencode-ai

2.0%

CLI tool for terminal-based AI coding assistance

created 4 months ago

updated 4 days ago

codemcp by ezyang

1.2%

Coding assistant for Claude Desktop, enabling code edits and testing

created 4 months ago

updated 1 month ago

Starred by

Lianmin Zheng(Author of SGLang),

Woosuk Kwon(Author of vLLM), and

4 more.

ao by pytorch

0.5%

PyTorch library for quantization and sparsity in training/inference

created 1 year ago

updated 18 hours ago

Starred by

Andrej Karpathy(Founder of Eureka Labs; Formerly at Tesla, OpenAI; Author of CS 231n),

Zhuohan Li(Author of vLLM), and

5 more.

torchtitan by pytorch

0.9%

PyTorch platform for generative AI model training research

created 1 year ago

updated 19 hours ago

Starred by

Lewis Tunstall(Researcher at Hugging Face),

Patrick von Platen(Core Contributor to Hugging Face Transformers and Diffusers), and

4 more.

torchtune by pytorch

0.2%

PyTorch library for LLM post-training and experimentation

created 1 year ago

updated 1 day ago

Starred by

Peter Norvig(Author of Artificial Intelligence: A Modern Approach; Research Director at Google),

Didier Lopes(Founder of OpenBB), and

14 more.

llm.c by karpathy

0.2%

27k

LLM training in pure C/CUDA, no PyTorch needed

created 1 year ago

updated 1 month ago

Starred by

Travis Fischer(Founder of Agentic),

Sebastian Pahl(Cofounder of Docker), and

2 more.

speechbrain by speechbrain

0.3%

10k

PyTorch toolkit for speech and text processing research

created 5 years ago

updated 5 days ago

Starred by

Chip Huyen(Author of AI Engineering, Designing Machine Learning Systems),

Zhuohan Li(Author of vLLM), and

4 more.

search_with_lepton by leptonai

0.1%

Conversational search engine demo

created 1 year ago

updated 1 month ago

Starred by

Michael Han(Cofounder of Unsloth),

Pietro Schirano(Founder of MagicPath), and

12 more.

text-generation-webui by oobabooga

0.2%

45k

Web UI for LLM text generation

created 2 years ago

updated 1 day ago

Starred by

Andrej Karpathy(Founder of Eureka Labs; Formerly at Tesla, OpenAI; Author of CS 231n),

Lianmin Zheng(Author of SGLang), and

12 more.

gpt-fast by pytorch-labs

0.1%

PyTorch text generation for efficient transformer inference

created 1 year ago

updated 3 months ago

Starred by

Chip Huyen(Author of AI Engineering, Designing Machine Learning Systems),

Omar Sanseviero(DevRel at Google DeepMind), and

3 more.

open_flamingo by mlfoundations

0.1%

Open-source framework for training large multimodal models

created 2 years ago

updated 11 months ago

Starred by

Tobi Lutke(Cofounder of Shopify),

Daniel Gross(Cofounder of Safe Superintelligence), and

19 more.

nanoGPT by karpathy

0.4%

43k

Minimalist repo for training/finetuning GPT models

created 2 years ago

updated 7 months ago

Starred by

Chenlin Meng(Cofounder of Pika),

Taranjeet Singh(Cofounder of Mem0), and

2 more.

disco-diffusion by alembics

0.0%

AI art and animation notebook

created 3 years ago

updated 2 years ago

Starred by

Chip Huyen(Author of AI Engineering, Designing Machine Learning Systems),

Travis Fischer(Founder of Agentic), and

4 more.

cleanlab by cleanlab

0.2%

11k

Data-centric AI package for ML with messy data

created 7 years ago

updated 3 weeks ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Travis Fischer(Founder of Agentic), and

11 more.

CLIP by openai

0.3%

30k

Image-text matching model for zero-shot prediction

created 4 years ago

updated 1 year ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Nat Friedman(Former CEO of GitHub), and

18 more.

minGPT by karpathy

0.2%

22k

Minimal PyTorch re-implementation for GPT training and inference

created 5 years ago

updated 11 months ago

Starred by

Chip Huyen(Author of AI Engineering, Designing Machine Learning Systems) and

Thomas Wolf(Cofounder of Hugging Face).

stable-baselines3 by DLR-RM

0.5%

11k

PyTorch library for reinforcement learning algorithm implementations

created 5 years ago

updated 1 week ago

Starred by

Luca Antiga(CTO of Lightning AI).

PhiFlow by tum-pbs

0.4%

Differentiable PDE solving framework for ML research

created 5 years ago

updated 20 hours ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Dan Abramov(Core Contributor to React), and

2 more.

jukebox by openai

0.1%

Generative model for music research paper

created 5 years ago

updated 1 year ago

Starred by

Luca Antiga(CTO of Lightning AI).

TensorRT by pytorch

0.3%

PyTorch compiler for NVIDIA GPUs using TensorRT

created 5 years ago

updated 20 hours ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Chip Huyen(Author of AI Engineering, Designing Machine Learning Systems), and

5 more.

pytorch3d by facebookresearch

0.2%

PyTorch3D is a PyTorch library for 3D deep learning research

created 5 years ago

updated 1 week ago

Starred by

Wei-Lin Chiang(Cofounder of LMArena),

Woosuk Kwon(Author of vLLM), and

8 more.

Megatron-LM by NVIDIA

0.4%

13k

Framework for training transformer models at scale

created 6 years ago

updated 23 hours ago

Starred by

Chip Huyen(Author of AI Engineering, Designing Machine Learning Systems).

jetson-inference by dusty-nv

0.2%

Vision DNN library for NVIDIA Jetson devices

created 9 years ago

updated 9 months ago

Starred by

Julien Chaumond(Cofounder of Hugging Face),

Jeff Hammerbacher(Cofounder of Cloudera), and

2 more.

naacl_transfer_learning_tutorial by huggingface

723

NLP transfer learning tutorial code

created 6 years ago

updated 5 years ago

texar-pytorch by asyml

746

PyTorch toolkit for NLP and text generation research

created 6 years ago

updated 3 years ago

Starred by

Andrej Karpathy(Founder of Eureka Labs; Formerly at Tesla, OpenAI; Author of CS 231n),

Omar Sanseviero(DevRel at Google DeepMind), and

11 more.

pytorch-image-models by huggingface

0.2%

35k

PyTorch image model collection with training, eval, and inference scripts

created 6 years ago

updated 16 hours ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Tom Brown(Cofounder of Anthropic), and

4 more.

XLM by facebookresearch

PyTorch code for cross-lingual language model pretraining (XLM/XLM-R)

created 6 years ago

updated 2 years ago

Starred by

Jeff Hammerbacher(Cofounder of Cloudera),

Sebastián Ramírez(Author of FastAPI, Typer, SQLModel, Asyncer), and

2 more.

spacy-course by explosion

Online course for advanced NLP with spaCy

created 6 years ago

updated 5 months ago

Starred by

Adam Paszke(Author of PyTorch) and

Thomas Wolf(Cofounder of Hugging Face).

pytorch-cpp-rl by Omegastick

524

RL framework using the PyTorch C++ frontend

created 6 years ago

updated 5 years ago

Starred by

Chip Huyen(Author of AI Engineering, Designing Machine Learning Systems),

Yang Song(Professor at Caltech; Research Scientist at OpenAI), and

15 more.

pytorch-lightning by Lightning-AI

0.1%

30k

Deep learning framework for pretraining, finetuning, and deploying AI models

created 6 years ago

updated 2 days ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Wei-Lin Chiang(Cofounder of LMArena), and

1 more.

BigGAN-PyTorch by ajbrock

PyTorch code for BigGAN image synthesis research paper

created 6 years ago

updated 2 years ago

Starred by

Travis Fischer(Founder of Agentic),

Amanpreet Singh(Cofounder of Contextual AI), and

5 more.

tch-rs by LaurentMazare

0.3%

Rust bindings for PyTorch C++ API (libtorch)

created 6 years ago

updated 1 day ago

Starred by

Thomas Wolf(Cofounder of Hugging Face).

gpt-2-Pytorch by graykode

Simple text generator with OpenAI GPT-2 PyTorch implementation

created 6 years ago

updated 6 years ago

Starred by

Stas Bekman(Author of Machine Learning Engineering Open Book; Research Engineer at Snowflake),

Lysandre Debut(Chief Open-Source Officer at Hugging Face), and

9 more.

pytext by facebookresearch

0.0%

NLP framework (deprecated, migrate to torchtext)

created 7 years ago

updated 2 years ago

Starred by

Omar Sanseviero(DevRel at Google DeepMind),

Zhiqiang Xie(Author of SGLang), and

7 more.

dgl by dmlc

0.0%

14k

Python package for deep learning on graphs

created 7 years ago

updated 1 day ago

Starred by

Chip Huyen(Author of AI Engineering, Designing Machine Learning Systems),

Ben Firshman(Cofounder of Replicate), and

5 more.

Made-With-ML by GokuMohandas

0.4%

41k

ML course for production-grade applications

created 6 years ago

updated 11 months ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Woosuk Kwon(Author of vLLM), and

9 more.

xla by pytorch

0.2%

PyTorch on XLA devices

created 6 years ago

updated 19 hours ago

Starred by

Logan Kilpatrick(Product Lead on Google AI Studio),

Chip Huyen(Author of AI Engineering, Designing Machine Learning Systems), and

2 more.

catalyst by catalyst-team

PyTorch framework for accelerated deep learning R&D

created 7 years ago

updated 1 month ago

Starred by

Chip Huyen(Author of AI Engineering, Designing Machine Learning Systems),

Elie Bursztein(Cybersecurity Lead at Google DeepMind), and

6 more.

mindsdb by mindsdb

0.5%

35k

AI query engine for federated data sources

created 7 years ago

updated 1 day ago

Starred by

Lilian Weng(Cofounder of Thinking Machines Lab),

Andrej Karpathy(Founder of Eureka Labs; Formerly at Tesla, OpenAI; Author of CS 231n), and

41 more.

transformers by huggingface

0.2%

148k

ML library for pretrained model inference and training

created 6 years ago

updated 10 hours ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Travis Fischer(Founder of Agentic), and

8 more.

tvm by apache

0.1%

12k

Compiler stack for deep learning systems

created 8 years ago

updated 15 hours ago

Starred by

Luca Antiga(CTO of Lightning AI) and

Jeff Hammerbacher(Cofounder of Cloudera).

pyhf by scikit-hep

288

HistFactory implementation for fitting/limit-setting/interval estimation

created 7 years ago

updated 2 weeks ago

Starred by

Jerry Tworek(VP of Research at OpenAI) and

Tom Brown(Cofounder of Anthropic).

vel by MillionIntegrals

277

Deep-learning research library for modular model construction

created 7 years ago

updated 2 years ago

yolov3 by ultralytics

0.1%

10k

Object detection in PyTorch > ONNX > CoreML > TFLite

created 7 years ago

updated 4 weeks ago

Starred by

Patrick von Platen(Core Contributor to Hugging Face Transformers and Diffusers) and

Lysandre Debut(Chief Open-Source Officer at Hugging Face).

espnet by espnet

0.3%

End-to-end speech processing toolkit for various speech tasks

created 7 years ago

updated 4 days ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Jeff Hammerbacher(Cofounder of Cloudera), and

4 more.

grokking-pytorch by Kaixhin

PyTorch guide with notes on usage, best practices, and debugging

created 7 years ago

updated 3 years ago

Starred by

Amanpreet Singh(Cofounder of Contextual AI),

Deshraj Yadav(Cofounder of Mem0), and

1 more.

lagom by zuoxingdong

376

PyTorch infrastructure for RL algorithm prototyping

created 7 years ago

updated 2 years ago

Starred by

Jeff Hammerbacher(Cofounder of Cloudera),

Amanpreet Singh(Cofounder of Contextual AI), and

1 more.

DALI by NVIDIA

0.0%

GPU-accelerated library for data pre-processing in deep learning

created 7 years ago

updated 1 day ago

Starred by

Lysandre Debut(Chief Open-Source Officer at Hugging Face),

Thomas Wolf(Cofounder of Hugging Face), and

2 more.

pytorch-openai-transformer-lm by huggingface

0.1%

PyTorch implementation of OpenAI's Transformer LM

created 7 years ago

updated 4 years ago

bindsnet by BindsNET

0.1%

PyTorch package for simulating spiking neural networks (SNNs)

created 7 years ago

updated 1 day ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Chip Huyen(Author of AI Engineering, Designing Machine Learning Systems), and

7 more.

higgsfield by higgsfield-ai

0.3%

ML framework for large model training and GPU orchestration

created 7 years ago

updated 1 year ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Jeff Hammerbacher(Cofounder of Cloudera), and

8 more.

apex by NVIDIA

0.1%

PyTorch extension for streamlined mixed precision & distributed training

created 7 years ago

updated 3 days ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Patrick von Platen(Core Contributor to Hugging Face Transformers and Diffusers), and

1 more.

tacotron2 by NVIDIA

0.0%

PyTorch implementation for text-to-speech synthesis

created 7 years ago

updated 1 year ago

Starred by

Sebastián Ramírez(Author of FastAPI, Typer, SQLModel, Asyncer).

training by mlcommons

0.1%

Reference implementations for MLPerf training benchmarks

created 7 years ago

updated 1 week ago

Starred by

Travis Fischer(Founder of Agentic).

TTS by mozilla

0.1%

10k

Deep learning library for text-to-speech generation

created 7 years ago

updated 1 year ago

jetson-reinforcement by dusty-nv

919

Deep RL GPU libraries for robotics on NVIDIA Jetson

created 9 years ago

updated 3 years ago

Starred by

Taranjeet Singh(Cofounder of Mem0) and

Thomas Wolf(Cofounder of Hugging Face).

sentiment-discovery by NVIDIA

0.1%

Language modeling and sentiment classification in PyTorch (deprecated, see Megatron-LM)

created 7 years ago

updated 5 years ago

Starred by

Jerry Tworek(VP of Research at OpenAI),

Andrew Trask(AI Researcher at Google DeepMind), and

2 more.

Rainbow by Kaixhin

0.1%

Deep reinforcement learning research paper implementation

created 7 years ago

updated 3 years ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

François Chollet(Author of Keras), and

20 more.

spaCy by explosion

0.2%

32k

NLP library for production applications

created 11 years ago

updated 2 months ago

Starred by

Jerry Tworek(VP of Research at OpenAI),

Guillaume Lample(Cofounder of Mistral), and

4 more.

pytorch-a2c-ppo-acktr-gail by ikostrikov

0.1%

PyTorch implementations of reinforcement learning algorithms

created 8 years ago

updated 3 years ago

Starred by

Lilian Weng(Cofounder of Thinking Machines Lab),

Aravind Srinivas(Cofounder of Perplexity), and

17 more.

fairseq by facebookresearch

0.1%

32k

Sequence modeling toolkit for translation, language modeling, and text generation research

created 8 years ago

updated 1 month ago

Starred by

Ben Mann(Cofounder of Anthropic) and

Deshraj Yadav(Cofounder of Mem0).

DeepRLHacks by williamFalcon

RL training hacks from Deep RL Bootcamp (2017)

created 8 years ago

updated 7 years ago

Starred by

Denis Yarats(Cofounder of Perplexity).

StarData by TorchCraft

573

StarCraft replay dataset for AI research

created 8 years ago

updated 3 years ago

Starred by

Jonathan Ragan-Kelley(Professor at MIT),

Tom Brown(Cofounder of Anthropic), and

4 more.

ELF by facebookresearch

Game research platform for real-time strategy (RTS) games

created 8 years ago

updated 3 years ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Ben Mann(Cofounder of Anthropic), and

2 more.

muniverse by unixpickle

365

RL environments for HTML5 games

created 8 years ago

updated 6 years ago

Starred by

Sebastian Raschka(Author of Build a Large Language Model From Scratch),

Jiayi Pan(Author of SWE-Gym; AI Researcher at UC Berkeley), and

1 more.

pytorch-grad-cam by jacobgil

0.3%

12k

PyTorch SDK for explainable AI in computer vision

created 8 years ago

updated 3 months ago

Starred by

Zhuohan Li(Author of vLLM).

rl_a3c_pytorch by dgriff777

570

PyTorch implementation of A3C for Atari games

created 8 years ago

updated 2 years ago

pytorch-REINFORCE by chingyaoc

266

PyTorch implementation of REINFORCE for control tasks

created 8 years ago

updated 8 years ago

Starred by

Jerry Tworek(VP of Research at OpenAI).

pytorch-rl by jingweiz

801

Deep RL research with PyTorch and Visdom

created 8 years ago

updated 5 years ago

Starred by

Adam Paszke(Author of PyTorch) and

Luca Antiga(CTO of Lightning AI).

pytorch-es by atgambardella

354

PyTorch implementation of Evolution Strategies for Markov Decision Processes

created 8 years ago

updated 8 years ago

Starred by

Philipp Schmid(DevRel at Google DeepMind),

Stas Bekman(Author of Machine Learning Engineering Open Book; Research Engineer at Snowflake), and

4 more.

the-incredible-pytorch by ritchieng

0.2%

12k

Curated list of PyTorch resources

created 8 years ago

updated 6 days ago

Starred by

Tim J. Baek(Founder of Open WebUI),

Stas Bekman(Author of Machine Learning Engineering Open Book; Research Engineer at Snowflake), and

6 more.

pytorch-tutorial by yunjey

0.1%

32k

PyTorch tutorial for deep learning researchers

created 8 years ago

updated 1 year ago

Starred by

Lilian Weng(Cofounder of Thinking Machines Lab),

Aravind Srinivas(Cofounder of Perplexity), and

4 more.

mujoco-py by openai

0.1%

Python bindings for MuJoCo physics engine

created 9 years ago

updated 1 year ago

Starred by

Lilian Weng(Cofounder of Thinking Machines Lab),

Aravind Srinivas(Cofounder of Perplexity), and

21 more.

faiss by facebookresearch

Starred by

Luca Antiga(CTO of Lightning AI).

pytorch-CortexNet by Atcold

367

PyTorch implementation of the CortexNet predictive model

created 8 years ago

updated 6 years ago

Starred by

Omar Sanseviero(DevRel at Google DeepMind),

Jeff Hammerbacher(Cofounder of Cloudera), and

4 more.

OpenNMT-py by OpenNMT

0.1%

PyTorch framework for neural machine translation and LLM experimentation

created 8 years ago

updated 4 months ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Guillaume Lample(Cofounder of Mistral), and

1 more.

pytorch-a3c by ikostrikov

0.1%

PyTorch implementation of A3C reinforcement learning algorithm

created 8 years ago

updated 5 years ago

Starred by

Amanpreet Singh(Cofounder of Contextual AI).

gym-starcraft by alibaba

532

OpenAI Gym environment for StarCraft AI research

created 8 years ago

updated 8 years ago

Starred by

Eugene Yan(AI Scientist at AWS),

Jeff Hammerbacher(Cofounder of Cloudera), and

7 more.

text by pytorch

0.1%

PyTorch library for NLP tasks

created 8 years ago

updated 1 day ago

Starred by

Lilian Weng(Cofounder of Thinking Machines Lab),

Jeff Hammerbacher(Cofounder of Cloudera), and

12 more.

examples by pytorch

0.1%

23k

PyTorch examples for diverse AI tasks

created 9 years ago

updated 6 days ago

Starred by

Jonathan Ragan-Kelley(Professor at MIT) and

Deshraj Yadav(Cofounder of Mem0).

openai-gemm by openai

382

GEMM kernels for single/half precision

created 8 years ago

updated 2 years ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Lilian Weng(Cofounder of Thinking Machines Lab), and

8 more.

ganhacks by soumith

0.1%

12k

GAN training tips and tricks

created 8 years ago

updated 3 years ago

Starred by

Boris Cherny(Creator of Claude Code; MTS at Anthropic),

Tom Brown(Cofounder of Anthropic), and

1 more.

Neural-Photo-Editor by ajbrock

Photo editor for generative neural network research

created 8 years ago

updated 8 years ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

François Chollet(Author of Keras), and

15 more.

gym by openai

0.1%

36k

RL toolkit for reinforcement learning algorithm development

created 9 years ago

updated 9 months ago

Starred by

Anton Osika(Cofounder of Lovable),

Luca Antiga(CTO of Lightning AI), and

1 more.

stat212b by joanbruna

DL course material (UC Berkeley, Spring 2016)

created 9 years ago

updated 7 years ago

Starred by

Quincy Larson(Founder of freeCodeCamp),

Anton Osika(Cofounder of Lovable), and

1 more.

speech_recognition by Uberi

0.1%

Python SDK for speech recognition, supporting online/offline engines and APIs

created 11 years ago

updated 2 months ago

Starred by

Deshraj Yadav(Cofounder of Mem0) and

Luca Antiga(CTO of Lightning AI).

awesome-torch by carpedm20

637

Curated list of Torch tutorials, projects, and communities

created 9 years ago

updated 7 years ago

Feedback? Help us improve.