⚡️ Lightning AI

Turn ideas into AI, Lightning fast. Creators of PyTorch Lightning, Lightning AI Studio, TorchMetrics, Fabric, Lit-GPT, Lit-LLaMA

README.md

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

pytorch-lightning pytorch-lightning Public

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

Python 26.6k 3.2k
torchmetrics torchmetrics Public

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

Python 1.9k 368
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.7k 493
litgpt litgpt Public

Hackable implementation of state-of-the-art open-source LLMs based on nanoGPT. Supports flash attention, 4-bit and 8-bit quantization, LoRA and LLaMA-Adapter fine-tuning, pre-training. Apache 2.0-l…

Python 5.4k 576

Repositories

lightning-thunder Public
Make PyTorch models Lightning fast! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors at once.

Python 772 Apache-2.0 43 4 (2 issues need help) 16 Updated Mar 27, 2024
lightning-Habana Public
Lightning support for Intel Habana accelerators.

Python 19 Apache-2.0 4 1 (1 issue needs help) 5 Updated Mar 27, 2024
pytorch-lightning Public
Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

Python 26,599 Apache-2.0 3,224 663 (144 issues need help) 58 Updated Mar 27, 2024
litgpt Public
Hackable implementation of state-of-the-art open-source LLMs based on nanoGPT. Supports flash attention, 4-bit and 8-bit quantization, LoRA and LLaMA-Adapter fine-tuning, pre-training. Apache 2.0-licensed.

Python 5,388 Apache-2.0 576 130 (1 issue needs help) 21 Updated Mar 27, 2024
ecosystem-ci Public
Automate issue discovery for your projects against Lightning nightly and releases.

Python 44 Apache-2.0 14 6 (2 issues need help) 8 Updated Mar 27, 2024
lightning-ui Public
Frontend for Lightning apps and home of the Design System

TypeScript 16 Apache-2.0 6 0 0 Updated Mar 26, 2024
torchmetrics Public
Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.

Python 1,877 Apache-2.0 368 69 (32 issues need help) 25 Updated Mar 26, 2024
litserve Public
Inference server for AI/ML models that is minimal and highly scalable.

Python 26 Apache-2.0 4 4 (1 issue needs help) 0 Updated Mar 26, 2024
utilities Public
Common Python utilities and GitHub Actions in Lightning Ecosystem

Python 41 Apache-2.0 12 3 (1 issue needs help) 1 Updated Mar 25, 2024
litdata Public
Blazingly fast, distributed streaming of training data from any cloud storage for training AI models

Python 155 Apache-2.0 7 11 (10 issues need help) 2 Updated Mar 25, 2024