Luca Antiga

Toolkit for linearizing PDFs for LLM datasets/training

Starred by

Jeff Hammerbacher(Cofounder of Cloudera) and

Dan Guido(Cofounder of Trail of Bits).

olmocr by allenai

1.4%

13k

created 10 months ago

Toy GRPO implementation for local RL training of LLMs

tiny-grpo by open-thought

1.1%

274

created 6 months ago

AI inference pipeline framework

Starred by

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

William Falcon(Founder of Lightning AI).

LitServe by Lightning-AI

1.0%

Tree-based speculative decoding algorithm (research paper)

Sequoia by Infini-AI-Lab

351

Open-source package for data-driven LLM application evaluation

continuous-eval by relari-ai

501

Starred by

William Falcon(Founder of Lightning AI),

Jeff Hammerbacher(Cofounder of Cloudera), and

1 more.

lightning-thunder by Lightning-AI

PyTorch compiler for model optimization via source-to-source transformation

SDK for scaling data transforms and optimizing datasets for fast AI training

updated 11 hours ago

Starred by

Jeff Hammerbacher(Cofounder of Cloudera).

litData by Lightning-AI

1.0%

516

Starred by

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

Shawn Wang(Editor of Latent Space), and

8 more.

mamba by state-spaces

Mamba SSM architecture for sequence modeling

16k

Starred by

Tobi Lutke(Cofounder of Shopify),

Simon Willison(Author of Django), and

3 more.

exllama by turboderp

Llama implementation for memory-efficient quantized weights

Starred by

Zhiqiang Xie(Author of SGLang),

Fabian Hedin(Cofounder of Lovable), and

21 more.

llama by meta-llama

Inference code for Llama 2 models (deprecated)

59k

PyTorch library for composable Transformer models

Starred by

Simon Willison(Author of Django),

Jeremy Howard(Cofounder of fast.ai), and

3 more.

curated-transformers by explosion

892

Genomic foundation model for long-range DNA sequence modeling

Starred by

Jeff Hammerbacher(Cofounder of Cloudera),

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

1 more.

hyena-dna by HazyResearch

704

Trainer for generative models, supporting diverse configurations

updated 3 months ago

naifu by Mikubill

326

updated 7 months ago

Starred by

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

Stefan van der Walt(Core Contributor to scientific Python ecosystem), and

7 more.

litgpt by Lightning-AI

LLM SDK for pretraining, finetuning, and deploying 20+ high-performance LLMs

13k

Open-source Python framework for holistic evaluation of foundation models

Starred by

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

Simon Willison(Author of Django), and

3 more.

helm by stanford-crfm

0.9%

created 3 years ago

Starred by

Nat Friedman(Former CEO of GitHub),

Alexey Milovidov(Cofounder of Clickhouse), and

5 more.

RedPajama-Data by togethercomputer

Dataset pipeline for training large language models

Compositional differentiable programming library for neuro-symbolic AI with LLMs

updated 7 months ago

Starred by

Phil Wang(Prolific Research Paper Implementer),

Eugene Yan(AI Scientist at AWS), and

1 more.

symbolicai by ExtensityAI

0.7%

Starred by

Tobi Lutke(Cofounder of Shopify),

Daniel Gross(Cofounder of Safe Superintelligence), and

19 more.

nanoGPT by karpathy

Minimalist repo for training/finetuning GPT models

43k

PyTorch transformer inference engine for GPU speedup

updated 7 months ago

Starred by

Amanpreet Singh(Cofounder of Contextual AI),

Philipp Schmid(DevRel at Google DeepMind), and

1 more.

kernl by ELS-RD

created 3 years ago

Starred by

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

Jeff Hammerbacher(Cofounder of Cloudera), and

4 more.

Transformers-Tutorials by NielsRogge

Transformer demos using Hugging Face, implemented in PyTorch

11k

created 4 years ago

Archived library for training Transformers with PyTorch Lightning

Starred by

Alexander Wettig(Author of SWE-bench, SWE-agent).

lightning-transformers by Lightning-Universe

609

created 4 years ago

Starred by

Anastasios Angelopoulos(Cofounder of LMArena).

torchio by TorchIO-project

TorchIO: Python package for medical image processing in AI

updated 19 hours ago

Starred by

Soumith Chintala(Author of PyTorch).

TensorRT by pytorch

PyTorch compiler for NVIDIA GPUs using TensorRT

Domain generalization resources (papers, datasets)

Domain-generalization by amber0309

839

updated 5 months ago

Starred by

Travis Fischer(Founder of Agentic),

Amanpreet Singh(Cofounder of Contextual AI), and

5 more.

tch-rs by LaurentMazare

Rust bindings for PyTorch C++ API (libtorch)

High-performance I/O system for large deep learning problems, strong PyTorch support

Starred by

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

Jeff Hammerbacher(Cofounder of Cloudera), and

7 more.

webdataset by webdataset

0.6%

Starred by

Wes McKinney(Author of Pandas),

Jeff Hammerbacher(Cofounder of Cloudera), and

1 more.

petastorm by uber

Data access library for deep learning model training/evaluation from Apache Parquet datasets

Starred by

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

mmagic by open-mmlab

AIGC toolbox for image/video editing and generation

Starred by

Patrick Kidger(Core Contributor to JAX ecosystem).

awesome-neural-ode by Zymrael

Resource list for neural differential equations and scientific ML

updated 10 months 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 15 hours ago

Starred by

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

Jeff Hammerbacher(Cofounder of Cloudera), and

1 more.

hummingbird by microsoft

Compiler for trained ML models into tensor computation

Starred by

Woosuk Kwon(Author of vLLM) and

Jeff Hammerbacher(Cofounder of Cloudera).

intel-extension-for-pytorch by intel

PyTorch extension for performance boost on Intel platforms

Curated list of open-source libraries for production ML

updated 5 days ago

Starred by

Shishir Patil(Author of BFCL, Gorilla),

Omar Sanseviero(DevRel at Google DeepMind), and

8 more.

awesome-production-machine-learning by EthicalML

0.6%

19k

Starred by

Soumith Chintala(Author of PyTorch).

PhiFlow by tum-pbs

Differentiable PDE solving framework for ML research

updated 17 hours ago

Starred by

Travis Fischer(Founder of Agentic),

Philipp Schmid(DevRel at Google DeepMind), and

9 more.

tokenizers by huggingface

Fast tokenizer library optimized for research and production

10k

updated 4 days 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

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

30k

HistFactory implementation for fitting/limit-setting/interval estimation

Starred by

Soumith Chintala(Author of PyTorch) and

Jeff Hammerbacher(Cofounder of Cloudera).

pyhf by scikit-hep

288

Starred by

Aravind Srinivas(Cofounder of Perplexity),

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

5 more.

pytorch3d by facebookresearch

PyTorch3D is a PyTorch library for 3D deep learning research

Starred by

Carol Willing(Core Contributor to CPython, Jupyter) and

Jeff Hammerbacher(Cofounder of Cloudera).

DiCE by interpretml

Counterfactual explanation SDK for ML models

Starred by

Patrick Kidger(Core Contributor to JAX ecosystem) and

Jeff Hammerbacher(Cofounder of Cloudera).

WeightWatcher by CalculatedContent

DNN analyzer for predicting model accuracy without training/test data

hamiltonian-nn by greydanus

Research paper code for Hamiltonian Neural Networks

477

updated 4 years ago

deep-high-resolution-net.pytorch by leoxiaobin

PyTorch SDK for human pose estimation

updated 11 months ago

info8010-deep-learning by glouppe

Deep learning lectures and assignments

updated 2 months 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

ML library for pretrained model inference and training

148k

ML lifecycle platform for managing experiments, models, and deployments

updated 7 hours ago

Starred by

Matei Zaharia(Cofounder of Databricks),

Wei-Lin Chiang(Cofounder of LMArena), and

4 more.

mlflow by mlflow

0.5%

21k

Starred by

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

Jeff Hammerbacher(Cofounder of Cloudera), and

1 more.

numpyro by pyro-ppl

Probabilistic programming library using JAX for GPU/TPU/CPU

PyTorch library for real-world deep reinforcement learning

Starred by

Thomas Wolf(Cofounder of Hugging Face).

machina by DeepX-inc

277

Model conversion tool for deep learning frameworks

updated 5 years ago

Starred by

Deshraj Yadav(Cofounder of Mem0) and

Adam Paszke(Author of PyTorch).

MMdnn by microsoft

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

NLP framework (deprecated, migrate to torchtext)

Starred by

Omar Sanseviero(DevRel at Google DeepMind),

Zhiqiang Xie(Author of SGLang), and

7 more.

dgl by dmlc

Python package for deep learning on graphs

14k

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Woosuk Kwon(Author of vLLM), and

9 more.

xla by pytorch

PyTorch on XLA devices

Starred by

Arvid Lunnemark(Cofounder of Cursor) and

Georgios Konstantopoulos(CTO, General Partner at Paradigm).

SEAL by microsoft

HE library for homomorphic encryption (additions/multiplications on encrypted data)

updated 4 months ago

katib by kubeflow

Kubernetes-native AutoML project

PyTorch framework for accelerated deep learning R&D

updated 5 days 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

Starred by

Chenlin Meng(Cofounder of Pika).

basic_reinforcement_learning by vmayoral

Introductory tutorial series for reinforcement learning techniques

PyTorch infrastructure for RL algorithm prototyping

Starred by

Amanpreet Singh(Cofounder of Contextual AI),

Deshraj Yadav(Cofounder of Mem0), and

1 more.

lagom by zuoxingdong

376

Starred by

Aravind Srinivas(Cofounder of Perplexity),

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

1 more.

tacotron2 by NVIDIA

PyTorch implementation for text-to-speech synthesis

Starred by

Aravind Srinivas(Cofounder of Perplexity),

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

7 more.

higgsfield by higgsfield-ai

ML framework for large model training and GPU orchestration

maddpg by openai

MADDPG research paper implementation

Starred by

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

Travis Fischer(Founder of Agentic), and

5 more.

prophet by facebook

Forecasting tool for time series data

19k

CLI tool for serving deep learning models from any ML/DL framework

multi-model-server by awslabs

Starred by

Jerry Tworek(VP of Research at OpenAI),

Andrew Trask(AI Researcher at Google DeepMind), and

2 more.

Rainbow by Kaixhin

Deep reinforcement learning research paper implementation

PyTorch DRL implementations for single/multi-agent environments

pytorch-DRL by ChenglongChen

595

Starred by

Aravind Srinivas(Cofounder of Perplexity),

François Chollet(Author of Keras), and

20 more.

spaCy by explosion

NLP library for production applications

32k

updated 2 months ago

Starred by

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

AI-Chip by basicmi

AI chip resource list

Starred by

Lilian Weng(Cofounder of Thinking Machines Lab),

Aravind Srinivas(Cofounder of Perplexity), and

17 more.

fairseq by facebookresearch

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

32k

Starred by

Tom Brown(Cofounder of Anthropic) and

Deshraj Yadav(Cofounder of Mem0).

SerpentAI by SerpentAI

Game agent framework for creating bots

Starred by

Jerry Tworek(VP of Research at OpenAI),

Guillaume Lample(Cofounder of Mistral), and

4 more.

pytorch-a2c-ppo-acktr-gail by ikostrikov

PyTorch implementations of reinforcement learning algorithms

Starred by

Travis Fischer(Founder of Agentic),

Lewis Tunstall(Researcher at Hugging Face), and

9 more.

umap by lmcinnes

Dimension reduction technique for visualization and general non-linear reduction

PyTorch implementation of the CortexNet predictive model

Starred by

Soumith Chintala(Author of PyTorch).

pytorch-CortexNet by Atcold

367

Lua-based toolkit for sequence-to-sequence learning

updated 6 years ago

Starred by

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

Andrey Vasnetsov(Cofounder of Qdrant), and

9 more.

fairseq-lua by facebookresearch

GAN resource list sorted by citations

Starred by

Amanpreet Singh(Cofounder of Contextual AI).

Delving-deep-into-GANs by GKalliatakis

853

Time-series database as a PostgreSQL extension for real-time analytics

updated 6 years ago

Starred by

Tim J. Baek(Founder of Open WebUI),

Jeffrey Morgan(Cofounder of Ollama), and

9 more.

timescaledb by timescale

0.5%

20k

PyTorch implementation of Evolution Strategies for Markov Decision Processes

Starred by

Adam Paszke(Author of PyTorch) and

Soumith Chintala(Author of PyTorch).

pytorch-es by atgambardella

354

updated 8 years 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

PyTorch tutorial for deep learning researchers

32k

Starred by

Lilian Weng(Cofounder of Thinking Machines Lab),

Aravind Srinivas(Cofounder of Perplexity), and

21 more.

faiss by facebookresearch

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

Curated list of PyTorch resources

12k

updated 6 days ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Guillaume Lample(Cofounder of Mistral), and

1 more.

pytorch-a3c by ikostrikov

PyTorch implementation of A3C reinforcement learning algorithm

updated 5 years 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

updated 5 days ago

deep_trader by deependersingla

RL agent for stock trading

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Lilian Weng(Cofounder of Thinking Machines Lab), and

8 more.

ganhacks by soumith

GAN training tips and tricks

12k

Starred by

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

Steve Sewell(Founder of Builder.io), and

6 more.

universe by openai

Deprecated platform for measuring/training AI across diverse applications

Speech recognition using WaveNet in TensorFlow

Starred by

Wei-Lin Chiang(Cofounder of LMArena),

Travis Fischer(Founder of Agentic), and

1 more.

speech-to-text-wavenet by buriburisuri

ANTs by ANTsX

C++ library for high-dimensional biomedical image analysis

created 12 years ago

updated 17 hours ago

Starred by

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

Jeff Hammerbacher(Cofounder of Cloudera), and

3 more.

tsfresh by blue-yonder

Python package for time series feature extraction

Starred by

Aravind Srinivas(Cofounder of Perplexity),

Ross Taylor(Cofounder of General Reasoning; Creator of Papers with Code), and

2 more.

pixel-cnn by openai

TensorFlow implementation for PixelCNN++ research paper

Deep reinforcement learning algorithms in Python, Keras-integrated

updated 5 years ago

Starred by

Aravind Srinivas(Cofounder of Perplexity),

François Chollet(Author of Keras), and

2 more.

keras-rl by keras-rl

Redis module for feed-forward neural networks as a native data type

Starred by

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

Tim Suchanek(Founder of expand.ai), and

5 more.

neural-redis by antirez

Starred by

Travis Fischer(Founder of Agentic) and

Jeff Hammerbacher(Cofounder of Cloudera).

serving by tensorflow

Serving system for machine learning models in production

Starred by

Aravind Srinivas(Cofounder of Perplexity),

François Chollet(Author of Keras), and

15 more.

gym by openai

RL toolkit for reinforcement learning algorithm development

36k

Deep reinforcement learning framework

updated 9 months ago

deer by VinF

488

DL course material (UC Berkeley, Spring 2016)

Starred by

Anton Osika(Cofounder of Lovable),

Luke Metz(AI Researcher at Thinking Machines Lab), and

1 more.

stat212b by joanbruna

Curated list of Torch tutorials, projects, and communities

Starred by

Deshraj Yadav(Cofounder of Mem0) and

Soumith Chintala(Author of PyTorch).

awesome-torch by carpedm20

637

Starred by

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

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

12 more.

CNTK by microsoft

Deep-learning toolkit for neural networks as directed computational graphs

18k

Starred by

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

Sourabh Bajaj(Cofounder of Uplimit), and

1 more.

awesome-deep-vision by kjw0612

Curated list of deep learning resources for computer vision

11k

created 10 years ago

Starred by

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

Aravind Srinivas(Cofounder of Perplexity), and

44 more.

tensorflow by tensorflow

191k

Open-source ML framework

C++ framework for reinforcement learning and planning problems

updated 6 hours ago

Starred by

Jerry Tworek(VP of Research at OpenAI).

AI-Toolbox by Svalorzen

659

JSON dataset of big data projects

updated 4 months ago

bigdata-ecosystem by zenkay

577

Starred by

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

Andrey Vasnetsov(Cofounder of Qdrant), and

11 more.

awesome-machine-learning by josephmisiti

Curated list of ML frameworks, libraries, and software

69k

Starred by

Philipp Schmid(DevRel at Google DeepMind),

Georgios Konstantopoulos(CTO, General Partner at Paradigm), and

2 more.

awesome-rust by rust-unofficial

Curated list of Rust code and resources

52k

Starred by

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

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

11 more.

redis by redis

Redis is a versatile data structure server, cache, and query engine

70k

created 16 years ago