⚡️ Lightning AI

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

    Python 30.9k 3.7k

  2. Machine learning metrics for distributed, scalable PyTorch applications.

    Python 2.4k 476

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

    Python 13.2k 1.4k

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

    Python 3.8k 275

  5. Speed up model training by fixing data loading.

    Python 580 90

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

    Python 1.4k 109

Repositories

Showing 10 of 25 repositories

  • torchmetrics Public

    Machine learning metrics for distributed, scalable PyTorch applications.

    Lightning-AI/torchmetrics’s past year of commit activity

  • pytorch-lightning Public

    Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

    Lightning-AI/pytorch-lightning’s past year of commit activity

  • litData Public

    Speed up model training by fixing data loading.

    Lightning-AI/litData’s past year of commit activity

  • utilities Public

    Common Python utilities and GitHub Actions in Lightning Ecosystem

    Lightning-AI/utilities’s past year of commit activity

  • litgpt Public

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

    Lightning-AI/litgpt’s past year of commit activity

  • LitLogger Public

    A minimal Python logger that tracks everything you try when building AI - metrics, prompts, models, etc, so you can see what changed and why.

    Lightning-AI/LitLogger’s past year of commit activity

    Python

    24

    Apache-2.0

    5 0 1

    Updated Mar 19, 2026

  • LitServe Public

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

    Lightning-AI/LitServe’s past year of commit activity

  • lightning-thunder Public

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

    Lightning-AI/lightning-thunder’s past year of commit activity

  • LitModels Public

    Save, load, host, and share AI model checkpoints without slowing down training. Host on Lightning AI or your own cloud with enterprise-grade access controls.

    Lightning-AI/LitModels’s past year of commit activity

  • litAI Public

    LLM router + minimal agent framework in one. Call any LLM API with OpenAI format. Unified billing, tools, retries, fallback, logging. Build agents and AI apps in pure Python. Full control. Zero magic.

    Lightning-AI/litAI’s past year of commit activity

    Python

    49

    Apache-2.0

    10 1 0

    Updated Jan 15, 2026