Yang Song

Lightweight library for evaluating language models

Starred by

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

Simon Willison(Author of Django), and

8 more.

simple-evals by openai

0.4%

created 1 year ago

updated 3 weeks ago

Starred by

Jeremy Howard(Cofounder of fast.ai),

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

3 more.

PiPPy by pytorch

PyTorch tool for pipeline parallelism

775

created 3 years ago

updated 11 months ago

Starred by

Georgios Konstantopoulos(CTO, General Partner at Paradigm),

Simon Willison(Author of Django), and

1 more.

consistencydecoder by openai

Improved decoding for stable diffusion VAEs

created 1 year ago

PyTorch library for vector quantization techniques

Starred by

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

vector-quantize-pytorch by lucidrains

0.4%

updated 1 week ago

Starred by

Patrick Kidger(Core Contributor to JAX ecosystem),

Travis Fischer(Founder of Agentic), and

7 more.

aim by aimhubio

Experiment tracker for AI model training runs

Hands-on puzzles for large language model training

Starred by

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

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

6 more.

LLM-Training-Puzzles by srush

1.0%

created 2 years ago

Starred by

Tobi Lutke(Cofounder of Shopify),

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

20 more.

guidance by guidance-ai

Guidance is a programming paradigm for steering LLMs

21k

created 2 years ago

updated 23 hours ago

Starred by

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

Travis Fischer(Founder of Agentic), and

2 more.

consistency_models by openai

PyTorch code for consistency models research paper

created 2 years ago

JAX library for neural networks and scientific computing

Starred by

Jeff Hammerbacher(Cofounder of Cloudera),

Jonathan Ragan-Kelley(Professor at MIT), and

1 more.

equinox by patrick-kidger

0.4%

Starred by

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

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

1 more.

glide-text2im by openai

Text-conditional image synthesis model from research paper

created 3 years ago

Starred by

Jeff Hammerbacher(Cofounder of Cloudera),

Thomas Wolf(Cofounder of Hugging Face), and

1 more.

numpyro by pyro-ppl

Probabilistic programming library using JAX for GPU/TPU/CPU

updated 2 weeks ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

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

2 more.

guided-diffusion by openai

Image synthesis codebase for diffusion models

Starred by

Elie Bursztein(Cybersecurity Lead at Google DeepMind),

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

4 more.

gpt-neo by EleutherAI

GPT-2/3-style model implementation using mesh-tensorflow

updated 3 years ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Nat Friedman(Former CEO of GitHub), and

18 more.

minGPT by karpathy

Minimal PyTorch re-implementation for GPT training and inference

22k

updated 11 months ago

Starred by

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

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

3 more.

taming-transformers by CompVis

Image synthesis research paper using transformers

Starred by

Aravind Srinivas(Cofounder of Perplexity),

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

11 more.

DeepSpeed by deepspeedai

Deep learning optimization library for distributed training and inference

40k

updated 21 hours ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Travis Fischer(Founder of Agentic), and

11 more.

CLIP by openai

Image-text matching model for zero-shot prediction

30k

Starred by

Robert Stojnic(Creator of Papers with Code).

ml-visuals by dair-ai

ML visuals for science communication

16k

updated 2 years ago

Starred by

Hiroshi Shibata(Core Contributor to Ruby),

Samuel Colvin(Author of Pydantic, Pydantic Logfire, PydanticAI), and

12 more.

streamlit by streamlit

SDK for rapidly building interactive data apps

41k

updated 18 hours ago

Starred by

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

Philipp Schmid(DevRel at Google DeepMind), and

15 more.

pytorch-lightning by Lightning-AI

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

30k

updated 2 days ago

Starred by

Chenlin Meng(Cofounder of Pika).

ncsn by ermongroup

Research paper implementation for generative modeling via score estimation

733

Starred by

Charlie Marsh(Founder of Astral; Author of Ruff, uv),

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

10 more.

flax by google

NN library for JAX, designed for flexibility in neural network research

updated 17 hours ago

Starred by

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

Nathan Lambert(AI Researcher at AI2).

tianshou by thu-ml

PyTorch RL library for algorithm development and application

Pytorch implementation of PixelCNN++ for generative image modeling

Starred by

Chenlin Meng(Cofounder of Pika).

pixel-cnn-pp by pclucas14

348

updated 4 years ago

Starred by

Jeff Hammerbacher(Cofounder of Cloudera),

Amanpreet Singh(Cofounder of Contextual AI), and

1 more.

DALI by NVIDIA

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

Starred by

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

George Hotz(Author of tinygrad; Founder of the tiny corp, comma.ai), and

10 more.

tqdm by tqdm

Progress bar for Python and CLI tools

30k

created 10 years ago

updated 2 months ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

François Chollet(Author of Keras), and

10 more.

horovod by horovod

Distributed training framework for TF, Keras, PyTorch, and MXNet

15k

created 8 years ago

updated 2 days ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Nathan Lambert(AI Researcher at AI2), and

1 more.

rlkit by rail-berkeley

RL algorithm collection implemented in PyTorch

Starred by

Lilian Weng(Cofounder of Thinking Machines Lab),

Aravind Srinivas(Cofounder of Perplexity), and

10 more.

tensor2tensor by tensorflow

Deprecated library for deep learning models/datasets, successor to Trax

16k

created 8 years ago

updated 2 years ago

Starred by

François Chollet(Author of Keras) and

Guillaume Lample(Cofounder of Mistral).

Arcade-Learning-Environment by Farama-Foundation

AI research platform for Atari 2600 game environments

created 13 years ago

updated 5 days ago

Starred by

Lilian Weng(Cofounder of Thinking Machines Lab),

Jeff Hammerbacher(Cofounder of Cloudera), and

12 more.

examples by pytorch

PyTorch examples for diverse AI tasks

23k

created 9 years ago

updated 6 days ago

Starred by

Lilian Weng(Cofounder of Thinking Machines Lab),

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

18 more.

awesome-deep-learning-papers by terryum

Curated list of seminal deep learning papers (2012-2016)

26k

created 9 years ago