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

  1. pytorch-lightning pytorch-lightning Public

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

    Python 26.8k 3.2k

  2. torchmetrics torchmetrics Public

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

    Python 1.9k 376

  3. lit-llama lit-llama Public

    Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Adapter fine-tuning, pre-training. Apache 2.0-licensed.

    Python 5.8k 498

  4. litgpt litgpt Public

    Pretrain, finetune, deploy 20+ LLMs on your own data. Uses state-of-the-art techniques: flash attention, FSDP, 4-bit, LoRA, and more.

    Python 6.3k 676

Repositories

Showing 10 of 22 repositories