Luca Antiga

CTO of Lightning AI

Starred Projects (110)

open-pi-zero by allenzren

Vision-language-action model re-implementation from "Physical Intelligence" paper

forresti:

Created 1 year ago

Updated 1 year ago

olmocr by allenai

Toolkit for linearizing PDFs for LLM datasets/training

chiphuyen:

hammer:

dguido:

omarsar:

Created 1 year ago

Updated 2 months ago

tiny-grpo by open-thought

Toy GRPO implementation for local RL training of LLMs

vincentweisser:

Created 1 year ago

Updated 1 year ago

LitServe by Lightning-AI

AI inference pipeline framework

chiphuyen:

luiscape:

omarsar:

williamFalcon:

Created 2 years ago

Updated 2 weeks ago

Sequoia by Infini-AI-Lab

Tree-based speculative decoding algorithm (research paper)

luiscape:

jiamings:

Created 2 years ago

Updated 1 year ago

continuous-eval by relari-ai

Open-source package for data-driven LLM application evaluation

rbehal:

Created 2 years ago

Updated 1 year ago

lightning-thunder by Lightning-AI

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

williamFalcon:

csarofeen:

hammer:

luiscape:

Created 2 years ago

Updated 6 days ago

litData by Lightning-AI

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

luiscape:

hammer:

Created 2 years ago

Updated 5 days ago

cutlass by NVIDIA

CUDA C++ and Python DSLs for high-performance linear algebra

pgarbacki:

pankajroark:

tridao:

chiphuyen:

Created 8 years ago

Updated 4 hours ago

mamba by state-spaces

Mamba SSM architecture for sequence modeling

geohot:

alexchen4ai:

luiscape:

zhiyuan8:

Created 2 years ago

Updated 2 weeks ago

exllama by turboderp

Llama implementation for memory-efficient quantized weights

tobi:

parano:

bryanhelmig:

simonw:

Created 3 years ago

Updated 2 years ago

llama by meta-llama

Inference code for Llama 2 models (deprecated)

froystig:

xiezhq-hermann:

fabhed:

borzunov:

Created 3 years ago

Updated 1 year ago

curated-transformers by explosion

PyTorch library for composable Transformer models

simonw:

jph00:

rromb:

millionintegrals:

Created 3 years ago

Updated 2 years ago

hyena-dna by HazyResearch

Genomic foundation model for long-range DNA sequence modeling

hammer:

chiphuyen:

JohannesHa:

lucidrains:

Created 2 years ago

Updated 1 year ago

naifu by Mikubill

Trainer for generative models, supporting diverse configurations

Created 3 years ago

Updated 1 year ago

litgpt by Lightning-AI

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

karpathy:

stefanv:

rasbt:

luiscape:

Created 3 years ago

Updated 3 weeks ago

helm by stanford-crfm

Open-source Python framework for holistic evaluation of foundation models

teetone:

hiyouga:

chiphuyen:

pgarbacki:

Created 4 years ago

Updated 4 days ago

RedPajama-Data by togethercomputer

Dataset pipeline for training large language models

nat:

alexey-milovidov:

CodeCreator:

chiphuyen:

Created 3 years ago

Updated 1 year ago

symbolicai by ExtensityAI

Compositional differentiable programming library for neuro-symbolic AI with LLMs

dsp:

lucidrains:

eugeneyan:

hammer:

Created 3 years ago

Updated 1 week ago

nanoGPT by karpathy

Minimalist repo for training/finetuning GPT models

tobi:

danielgross:

chiphuyen:

ankane:

Created 3 years ago

Updated 6 months ago

dl-fundamentals by Lightning-AI

Deep learning fundamentals and practical applications

Created 3 years ago

Updated 2 years ago

kernl by ELS-RD

PyTorch transformer inference engine for GPU speedup

borzunov:

casper-hansen:

apsdehal:

philschmid:

Created 3 years ago

Updated 3 months ago

lightning-transformers by Lightning-Universe

Archived library for training Transformers with PyTorch Lightning

soldni:

CodeCreator:

luiscape:

Created 5 years ago

Updated 3 years ago

torchio by TorchIO-project

TorchIO: Python package for medical image processing in AI

omarsar:

aangelopoulos:

Created 6 years ago

Updated 2 weeks ago

TensorRT by pytorch

PyTorch compiler for NVIDIA GPUs using TensorRT

luiscape:

codekansas:

soumith:

Created 6 years ago

Updated 16 hours ago

Domain-generalization by amber0309

Domain generalization resources (papers, datasets)

Created 7 years ago

Updated 1 year ago

tch-rs by LaurentMazare

Rust bindings for PyTorch C++ API (libtorch)

ekzhang:

sxyu:

transitive-bullshit:

parano:

Created 7 years ago

Updated 1 week ago

webdataset by webdataset

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

soldni:

chiphuyen:

hammer:

theophilegervet:

Created 6 years ago

Updated 3 months ago

petastorm by uber

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

jn2clark:

omarsar:

luiscape:

wesm:

Created 8 years ago

Updated 4 months ago

mmagic by open-mmlab

AIGC toolbox for image/video editing and generation

chiphuyen:

agermanidis:

wsxiaoys:

Created 6 years ago

Updated 1 year ago

awesome-neural-ode by Zymrael

Resource list for neural differential equations and scientific ML

ablattmann:

rromb:

patrick-kidger:

Created 6 years ago

Updated 1 year ago

streamlit by streamlit

SDK for rapidly building interactive data apps

kellyamanda:

treuille:

hsbt:

samuelcolvin:

Created 6 years ago

Updated 9 hours ago

hummingbird by microsoft

Compiler for trained ML models into tensor computation

tgaddair:

chiphuyen:

parano:

ankane:

Created 6 years ago

Updated 10 months ago

intel-extension-for-pytorch by intel

PyTorch extension for performance boost on Intel platforms

WoosukKwon:

hammer:

Created 6 years ago

Updated 1 month ago

awesome-production-machine-learning by EthicalML

Curated list of open-source libraries for production ML

antiagainst:

ShishirPatil:

ogabrielluiz:

osanseviero:

Created 7 years ago

Updated 1 day ago

simpletransformers by ThilinaRajapakse

Rapid NLP task implementation

lysandrejik:

shizhediao:

gakonst:

ogabrielluiz:

Created 6 years ago

Updated 9 months ago

PhiFlow by tum-pbs

Differentiable PDE solving framework for ML research

Edward-Sun:

soumith:

rromb:

Created 6 years ago

Updated 2 weeks ago

tokenizers by huggingface

Fast tokenizer library optimized for research and production

clmnt:

hammer:

syrusakbary:

transitive-bullshit:

Created 6 years ago

Updated 2 days ago

pytorch-lightning by Lightning-AI

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

albertfgu:

soldni:

omarsar:

zhangce:

Created 7 years ago

Updated 4 days ago

pyhf by scikit-hep

HistFactory implementation for fitting/limit-setting/interval estimation

mattjj:

soumith:

hammer:

Created 8 years ago

Updated 6 days ago

pytorch3d by facebookresearch

PyTorch3D is a PyTorch library for 3D deep learning research

aravindsrinivas:

chiphuyen:

hiyouga:

gkioxari:

Created 6 years ago

Updated 1 week ago

DiCE by interpretml

Counterfactual explanation SDK for ML models

willingc:

hammer:

Created 7 years ago

Updated 10 months ago

WeightWatcher by CalculatedContent

DNN analyzer for predicting model accuracy without training/test data

patrick-kidger:

hammer:

Created 7 years ago

Updated 1 week ago

vedo by marcomusy

Python module for 3D scientific analysis and visualization

Created 8 years ago

Updated 3 weeks ago

hamiltonian-nn by greydanus

Research paper code for Hamiltonian Neural Networks

rromb:

Created 7 years ago

Updated 5 years ago

deep-high-resolution-net.pytorch by leoxiaobin

PyTorch SDK for human pose estimation

youkaichao:

Created 7 years ago

Updated 1 year ago

tract by sonos

Tiny, self-contained inference engine for diverse hardware and modalities

AlexCheema:

gakonst:

binarybana:

kerollmops:

Created 8 years ago

Updated 2 days ago

transformers by huggingface

ML library for pretrained model inference and training

clmnt:

lilianweng:

karpathy:

tjbck:

Created 7 years ago

Updated 5 hours ago

mlflow by mlflow

ML lifecycle platform for managing experiments, models, and deployments

mateiz:

jn2clark:

infwinston:

thinkall:

Created 8 years ago

Updated 4 hours ago

numpyro by pyro-ppl

Probabilistic programming library using JAX for GPU/TPU/CPU

omarsar:

froystig:

yang-song:

hammer:

Created 7 years ago

Updated 1 day ago

machina by DeepX-inc

PyTorch library for real-world deep reinforcement learning

thomwolf:

Created 8 years ago

Updated 6 years ago

MMdnn by microsoft

Model conversion tool for deep learning frameworks

deshraj:

parano:

apaszke:

Created 8 years ago

Updated 9 months ago

pytext by facebookresearch

NLP framework (deprecated, migrate to torchtext)

stas00:

lysandrejik:

chiphuyen:

codekansas:

Created 7 years ago

Updated 3 years ago

dgl by dmlc

Python package for deep learning on graphs

osanseviero:

jiamings:

xiezhq-hermann:

parasj:

Created 8 years ago

Updated 9 months ago

xla by pytorch

PyTorch on XLA devices

aravindsrinivas:

WoosukKwon:

Jiayi-Pan:

comaniac:

Created 7 years ago

Updated 2 weeks ago

SEAL by microsoft

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

arvid220u:

ankane:

rxin:

gitbuda:

Created 7 years ago

Updated 3 weeks ago

SynapseML by microsoft

Distributed ML library for large-scale data processing

hammer:

AntonOsika:

infwinston:

deshraj:

Created 9 years ago

Updated 3 days ago

server by triton-inference-server

AI model inference serving optimized for cloud and edge

hammer:

tjbck:

Edward-Sun:

jn2clark:

Created 7 years ago

Updated 2 days ago

katib by kubeflow

Kubernetes-native AutoML project

luiscape:

Created 8 years ago

Updated 3 days ago

catalyst by catalyst-team

PyTorch framework for accelerated deep learning R&D

logankilpatrick:

parasj:

chiphuyen:

luiscape:

Created 7 years ago

Updated 11 months ago

basic_reinforcement_learning by vmayoral

Introductory tutorial series for reinforcement learning techniques

chenlin9:

vincentweisser:

Created 10 years ago

Updated 2 years ago

lagom by zuoxingdong

PyTorch infrastructure for RL algorithm prototyping

apsdehal:

jachiam:

deshraj:

soumith:

Created 8 years ago

Updated 3 years ago

pro_gan_pytorch by akanimax

PyTorch implementation of Progressive GANs for image synthesis

soumith:

Created 8 years ago

Updated 2 years ago

tacotron2 by NVIDIA

PyTorch implementation for text-to-speech synthesis

aravindsrinivas:

patrickvonplaten:

jongwook:

soumith:

Created 8 years ago

Updated 1 year ago

higgsfield by higgsfield-ai

ML framework for large model training and GPU orchestration

aravindsrinivas:

luiscape:

pgarbacki:

chiphuyen:

Created 8 years ago

Updated 2 years ago

pytorch-kaldi by mravanelli

Speech recognition toolkit bridging PyTorch and Kaldi

soumith:

tridao:

Created 8 years ago

Updated 4 years ago

maddpg by openai

MADDPG research paper implementation

Created 8 years ago

Updated 2 years ago

prophet by facebook

Forecasting tool for time series data

bcherny:

transitive-bullshit:

parano:

JohannesHa:

Created 9 years ago

Updated 2 weeks ago

multi-model-server by awslabs

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

simon-mo:

Created 8 years ago

Updated 2 years ago

Rainbow by Kaixhin

Deep reinforcement learning research paper implementation

evhub:

millionintegrals:

jwyang:

iamtrask:

Created 8 years ago

Updated 4 years ago

pytorch-DRL by ChenglongChen

PyTorch DRL implementations for single/multi-agent environments

Created 8 years ago

Updated 8 years ago

spaCy by explosion

NLP library for production applications

aravindsrinivas:

fchollet:

nirga:

jn2clark:

Created 12 years ago

Updated 6 days ago

AI-Chip by basicmi

AI chip resource list

stas00:

casper-hansen:

forresti:

antiagainst:

Created 8 years ago

Updated 1 year ago

fairseq by facebookresearch

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

lilianweng:

aravindsrinivas:

tjbck:

pathak22:

Created 8 years ago

Updated 7 months ago

SerpentAI by SerpentAI

Game agent framework for creating bots

cvalenzuela:

nottombrown:

deshraj:

Created 9 years ago

Updated 3 years ago

pytorch-a2c-ppo-acktr-gail by ikostrikov

PyTorch implementations of reinforcement learning algorithms

theophilegervet:

jachiam:

rwightman:

millionintegrals:

Created 8 years ago

Updated 4 years ago

umap by lmcinnes

Dimension reduction technique for visualization and general non-linear reduction

transitive-bullshit:

lewtun:

ankane:

patrick-kidger:

Created 9 years ago

Updated 3 weeks ago

pytorch-CortexNet by Atcold

PyTorch implementation of the CortexNet predictive model

soumith:

Created 9 years ago

Updated 7 years ago

rxdb by pubkey

Reactive, local-first NoSQL database for JavaScript

hammer:

ekzhang:

jeffchuber:

swyxio:

Created 9 years ago

Updated 1 day ago

fairseq-lua by facebookresearch

Lua-based toolkit for sequence-to-sequence learning

bcherny:

generall:

thomwolf:

vincentweisser:

Created 9 years ago

Updated 4 years ago

Delving-deep-into-GANs by GKalliatakis

GAN resource list sorted by citations

apsdehal:

huybery:

Created 9 years ago

Updated 7 years ago

timescaledb by timescale

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

quincylarson:

mfreed:

tjbck:

amin3141:

Created 9 years ago

Updated 4 hours ago

pytorch-es by atgambardella

PyTorch implementation of Evolution Strategies for Markov Decision Processes

jekbradbury:

apaszke:

soumith:

Created 9 years ago

Updated 8 years ago

pytorch-tutorial by yunjey

PyTorch tutorial for deep learning researchers

tjbck:

bryanhelmig:

stas00:

haotian-liu:

Created 9 years ago

Updated 2 years ago

faiss by facebookresearch

Similarity search library for dense vectors

lilianweng:

aravindsrinivas:

hsbt:

khou22:

Created 9 years ago

Updated 2 days ago

the-incredible-pytorch by ritchieng

Curated list of PyTorch resources

ogabrielluiz:

philschmid:

stas00:

john-b-yang:

Created 9 years ago

Updated 1 month ago

pytorch-a3c by ikostrikov

PyTorch implementation of A3C reinforcement learning algorithm

aravindsrinivas:

glample:

jekbradbury:

soumith:

Created 9 years ago

Updated 6 years ago

examples by pytorch

PyTorch examples for diverse AI tasks

lilianweng:

hammer:

tjbck:

ari-holtzman:

Created 9 years ago

Updated 8 months ago

deep_trader by deependersingla

RL agent for stock trading

Created 10 years ago

Updated 8 years ago

ganhacks by soumith

GAN training tips and tricks

aravindsrinivas:

lilianweng:

ebursztein:

stefanv:

Created 9 years ago

Updated 4 years ago

universe by openai

Deprecated platform for measuring/training AI across diverse applications

Jiayi-Pan:

codekansas:

junxiaosong:

evhub:

Created 9 years ago

Updated 8 years ago

speech-to-text-wavenet by buriburisuri

Speech recognition using WaveNet in TensorFlow

parano:

cvalenzuela:

infwinston:

transitive-bullshit:

Created 9 years ago

Updated 4 years ago

ANTs by ANTsX

C++ library for high-dimensional biomedical image analysis

Created 13 years ago

Updated 12 hours ago

tsfresh by blue-yonder

Python package for time series feature extraction

bcherny:

hammer:

rodrigosnader:

tunguz:

Created 9 years ago

Updated 6 months ago

pixel-cnn by openai

TensorFlow implementation for PixelCNN++ research paper

aravindsrinivas:

evhub:

jinze1994:

omarsar:

Created 10 years ago

Updated 6 years ago

keras-rl by keras-rl

Deep reinforcement learning algorithms in Python, Keras-integrated

aravindsrinivas:

fchollet:

lysandrejik:

vincentweisser:

Created 10 years ago

Updated 2 years ago

neural-redis by antirez

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

bcherny:

luiscape:

timsuchanek:

simonw:

Created 9 years ago

Updated 7 years ago

serving by tensorflow

Serving system for machine learning models in production

salanki:

transitive-bullshit:

jamesr66a:

cvalenzuela:

Created 10 years ago

Updated 1 week ago

gym by openai

RL toolkit for reinforcement learning algorithm development

aravindsrinivas:

fchollet:

bcherny:

ValentaTomas:

Created 10 years ago

Updated 1 month ago

deer by VinF

Deep reinforcement learning framework

pathak22:

agermanidis:

Created 10 years ago

Updated 11 months ago

awesome-torch by carpedm20

Curated list of Torch tutorials, projects, and communities

deshraj:

jwyang:

soumith:

Created 10 years ago

Updated 8 years ago

CNTK by microsoft

Deep-learning toolkit for neural networks as directed computational graphs

bcherny:

tiangolo:

luiscape:

parano:

Created 10 years ago

Updated 3 years ago

awesome-deep-vision by kjw0612

Curated list of deep learning resources for computer vision

alexandr:

bcherny:

huybery:

sb2nov:

Created 11 years ago

Updated 2 years ago

tensorflow by tensorflow

Open-source ML framework

norvig:

aravindsrinivas:

karpathy:

bcherny:

Created 10 years ago

Updated 4 hours ago

airflow by apache

Workflow orchestration platform for programmatic scheduling and monitoring

ValentaTomas:

transitive-bullshit:

charliermarsh:

jspahrsummers:

Created 11 years ago

Updated 4 hours ago

AI-Toolbox by Svalorzen

C++ framework for reinforcement learning and planning problems

millionintegrals:

artidoro:

Created 12 years ago

Updated 1 year ago

bigdata-ecosystem by zenkay

JSON dataset of big data projects

Created 11 years ago

Updated 4 years ago

awesome-machine-learning by josephmisiti

Curated list of ML frameworks, libraries, and software

bcherny:

vincentweisser:

generall:

parano:

Created 12 years ago

Updated 1 week ago

awesome-rust by rust-unofficial

Curated list of Rust code and resources

philschmid:

gakonst:

peakji:

spartee:

Created 12 years ago

Updated 1 day ago

redis by redis

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

mikeyk:

patrickvonplaten:

nirga:

chiphuyen:

Created 17 years ago

Updated 4 days ago

Feedback? Help us improve.