⚡️ Lightning AI

All

22 repositories

litgpt
Public
20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.
ai deep-learning artificial-intelligence large-language-models llm llms llm-inference
Python
•
Apache License 2.0
•1.2k•12k•267•12•Updated May 6, 2025May 6, 2025
lightning-thunder
Public
Thunder gives you PyTorch models superpowers for training and inference. Unlock out-of-the-box optimizations for performance, memory and parallelism, or roll out your own.
Python
•
Apache License 2.0
•93•1.3k•326•59•Updated May 6, 2025May 6, 2025
pytorch-lightning
Public
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
python data-science machine-learning ai deep-learning pytorch artificial-intelligence
Python
•
Apache License 2.0
•3.5k•29k•931•62•Updated May 6, 2025May 6, 2025
torchmetrics
Public
Machine learning metrics for distributed, scalable PyTorch applications.
python data-science machine-learning deep-learning metrics pytorch analyses
Python
•
Apache License 2.0
•431•2.3k•71•14•Updated May 6, 2025May 6, 2025
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.
model sharing checkpoint fine-tuning
Python
•
Apache License 2.0
•4•22•2•0•Updated May 6, 2025May 6, 2025
LitServe
Public
Deploy high-performance AI models and inference pipelines on FastAPI with built-in batching, streaming and more.
api web ai deep-learning rest-api artificial-intelligence developer-tools serving fastapi
Python
•
Apache License 2.0
•200•3.1k•22•4•Updated May 6, 2025May 6, 2025
utilities
Public
Common Python utilities and GitHub Actions in Lightning Ecosystem
ci-cd developer-tools github-actions
Python
•
Apache License 2.0
•16•56•2•0•Updated May 6, 2025May 6, 2025
ecosystem-ci
Public archive
Automate issue discovery for your projects against Lightning nightly and releases.
lightweight deep-learning integration-testing ci-cd compatibility-testing python
Python
•
Apache License 2.0
•16•47•6•8•Updated May 6, 2025May 6, 2025
litData
Public
Transform datasets at scale. Optimize datasets for fast AI model training.
Python
•
Apache License 2.0
•65•468•49•7•Updated May 3, 2025May 3, 2025
tutorials
Public
Collection of Pytorch lightning tutorial form as rich scripts automatically transformed to ipython notebooks.
machine-learning lightning deep-learning jupyter-notebook tutorials notebooks python-scripts
Python
•
Apache License 2.0
•80•314•12•2•Updated May 1, 2025May 1, 2025
hello-studio
Public
Starter projects for Lightning Studios
Apache License 2.0
•0•0•0•0•Updated Apr 15, 2025Apr 15, 2025
lightning-Habana
Public
Lightning support for Intel Habana accelerators.
python machine-learning deep-learning pytorch
Python
•
Apache License 2.0
•8•27•1•12•Updated Apr 7, 2025Apr 7, 2025
probot
Public
TypeScript
•
ISC License
•13•9•0•1•Updated Mar 28, 2025Mar 28, 2025
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
•
Apache License 2.0
•522•6.1k•109•11•Updated Sep 6, 2024Sep 6, 2024
dl-fundamentals
Public
Deep Learning Fundamentals -- Code material and exercises
python ai machine-learning deep-learning pytorch
Jupyter Notebook
•193•374•3•0•Updated Feb 28, 2024Feb 28, 2024
lightning-Graphcore
Public archive
Python
•
Apache License 2.0
•4•8•2•6•Updated Feb 1, 2024Feb 1, 2024
.github
Public
0•0•0•0•Updated Dec 24, 2023Dec 24, 2023
Lightning-multinode-templates
Public
Multinode templates for Pytorch Lightning
Python
•
Apache License 2.0
•1•7•1•1•Updated Nov 26, 2023Nov 26, 2023
deep-learning-project-template
Public template
Pytorch Lightning code guideline for conferences
Python
•
Apache License 2.0
•276•1.3k•11•3•Updated Sep 20, 2023Sep 20, 2023
lm-evaluation-harness
Public archive
A framework for few-shot evaluation of autoregressive language models.
Python
•
MIT License
•2.4k•3•0•0•Updated Jun 5, 2023Jun 5, 2023
forked-pdb
Public archive
Python pdb for multiple processes
Python
•
Apache License 2.0
•6•40•0•0•Updated Nov 5, 2022Nov 5, 2022
engineering-class
Public
Lightning Bits: Engineering for Researchers repo
Python
•17•132•0•0•Updated Aug 29, 2022Aug 29, 2022