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 28.5k 3.4k

2. torchmetrics torchmetrics Public

   Machine learning metrics for distributed, scalable PyTorch applications.

   Python 2.1k 408

3. litgpt litgpt Public

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

   Python 10.8k 1.1k

4. LitServe LitServe Public

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

   Python 2.5k 160

5. litdata litdata Public

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

   Python 375 42

6. lightning-thunder lightning-thunder Public

   Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors at once; across one or thousands of GPUs.

   Python 1.2k 80