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 424

3. litgpt litgpt Public

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

   Python 11.8k 1.2k

4. LitServe LitServe Public

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

   Python 3k 192

5. litdata litdata Public

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

   Python 427 59

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.3k 90