Skip to content
@Lightning-AI

⚡️ Lightning AI

Turn ideas into AI, Lightning fast. Creators of PyTorch Lightning, Lightning AI Studio, TorchMetrics, Fabric, Lit-GPT, Lit-LLaMA

Welcome to Lightning AI ⚡

The all-in-one platform for AI development. Code together. Prototype. Train. Scale. Serve. From your browser - with zero setup. From the creators of PyTorch Lightning.

Pinned Loading

  1. pytorch-lightning pytorch-lightning Public

    Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

    Python 28k 3.4k

  2. torchmetrics torchmetrics Public

    Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.

    Python 2.1k 400

  3. litgpt litgpt Public

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

    Python 9.8k 982

  4. LitServe LitServe Public

    Lightning-fast serving engine for AI models. Flexible. Easy. Enterprise-scale.

    Python 2.1k 129

  5. litdata litdata Public

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

    Python 321 38

  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.1k 72

Repositories

Showing 10 of 21 repositories