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 ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

    Python 30.6k 3.6k

  2. torchmetrics torchmetrics Public

    Machine learning metrics for distributed, scalable PyTorch applications.

    Python 2.4k 468

  3. litgpt litgpt Public

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

    Python 13.1k 1.4k

  4. LitServe LitServe Public

    A minimal Python framework for building custom AI inference servers with full control over logic, batching, and scaling.

    Python 3.7k 263

  5. litData litData Public

    Speed up model training by fixing data loading.

    Python 566 81

  6. lightning-thunder lightning-thunder Public

    PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily write your own.

    Python 1.4k 111

Repositories

Showing 10 of 25 repositories