Pinned Loading

1. Lightning-AI/pytorch-lightning Lightning-AI/pytorch-lightning Public

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

   Python 28.4k 3.4k

2. Lightning-AI/litgpt Lightning-AI/litgpt Public

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

   Python 10.8k 1.1k

3. Lightning-AI/torchmetrics Lightning-AI/torchmetrics Public

   Machine learning metrics for distributed, scalable PyTorch applications.

   Python 2.1k 409

4. Lightning-AI/LitServe Lightning-AI/LitServe Public

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

   Python 2.5k 158

5. pytorch/pytorch pytorch/pytorch Public

   Tensors and Dynamic neural networks in Python with strong GPU acceleration

   Python 84.4k 22.7k

6. test-tube test-tube Public

   Python library to easily log experiments and parallelize hyperparameter search for neural networks

   JavaScript 736 74