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

  2. torchmetrics torchmetrics Public

    Machine learning metrics for distributed, scalable PyTorch applications.

    Python 2.3k 466

  3. litgpt litgpt Public

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

    Python 12.8k 1.3k

  4. LitServe LitServe Public

    Build custom inference engines for models, agents, multi-modal systems, RAG, pipelines and more.

    Python 3.6k 244

  5. litData litData Public

    Speed up model training by fixing data loading.

    Python 551 80

  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 107

Repositories

Showing 10 of 24 repositories