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 multiple GPUs, TPUs with zero code changes.

    Python 28.4k 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.7k 1.1k

  4. LitServe LitServe Public

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

    Python 2.5k 157

  5. litdata litdata Public

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

    Python 367 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

Repositories

Showing 10 of 22 repositories