Pinned Loading

1. pytorch-lightning pytorch-lightning Public

   Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.

   Python 29.5k 3.5k

2. torchmetrics torchmetrics Public

   Machine learning metrics for distributed, scalable PyTorch applications.

   Python 2.3k 432

3. litgpt litgpt Public

   20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.

   Python 12.1k 1.2k

4. LitServe LitServe Public

   Deploy high-performance AI models and inference pipelines on FastAPI with built-in batching, streaming and more.

   Python 3.1k 206

5. litData litData Public

   Transform datasets at scale. Optimize datasets for fast AI model training.

   Python 476 64

6. lightning-thunder lightning-thunder Public

   Thunder gives you PyTorch models superpowers for training and inference. Unlock out-of-the-box optimizations for performance, memory and parallelism, or roll out your own.

   Python 1.3k 93