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.2k 3.5k

2. torchmetrics torchmetrics Public

   Machine learning metrics for distributed, scalable PyTorch applications.

   Python 2.2k 426

3. litgpt litgpt Public

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

   Python 11.9k 1.2k

4. LitServe LitServe Public

   Lightning-fast serving engine for any AI model of any size. Flexible. Easy. Enterprise-scale.

   Python 3k 195

5. litdata litdata Public

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

   Python 435 59

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 91