⚡️ Lightning AI

All

20 repositories

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.4k•29k•855•65•Updated Feb 3, 2025Feb 3, 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
•419•2.2k•91•23•Updated Feb 1, 2025Feb 1, 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•297•12•3•Updated Feb 1, 2025Feb 1, 2025
lightning-Habana
Public
Lightning support for Intel Habana accelerators.
python machine-learning deep-learning pytorch
Python
•
Apache License 2.0
•8•25•1•12•Updated Feb 1, 2025Feb 1, 2025
lightning-thunder
Public
Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors at once; across one or thousands of GPUs.
Python
•
Apache License 2.0
•87•1.3k•295•45•Updated Feb 1, 2025Feb 1, 2025
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.1k•11k•249•12•Updated Jan 31, 2025Jan 31, 2025
utilities
Public
Common Python utilities and GitHub Actions in Lightning Ecosystem
ci-cd developer-tools github-actions
Python
•
Apache License 2.0
•16•51•4•2•Updated Jan 31, 2025Jan 31, 2025
ecosystem-ci
Public
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•45•6•9•Updated Jan 27, 2025Jan 27, 2025
litdata
Public
Transform datasets at scale. Optimize datasets for fast AI model training.
Python
•
Apache License 2.0
•51•409•57•4•Updated Jan 24, 2025Jan 24, 2025
LitServe
Public
Lightning-fast serving engine for any AI model of any size. Flexible. Easy. Enterprise-scale.
api web ai deep-learning rest-api artificial-intelligence developer-tools serving fastapi
Python
•
Apache License 2.0
•181•2.8k•20•5•Updated Jan 21, 2025Jan 21, 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
•517•6k•108•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
•190•365•3•0•Updated Feb 28, 2024Feb 28, 2024
lightning-Graphcore
Public archive
Python
•
Apache License 2.0
•4•7•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
probot
Public
TypeScript
•
ISC License
•13•9•0•1•Updated Oct 9, 2023Oct 9, 2023
deep-learning-project-template
Public template
Pytorch Lightning code guideline for conferences
Python
•
Apache License 2.0
•273•1.2k•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
•2k•3•0•0•Updated Jun 5, 2023Jun 5, 2023
forked-pdb
Public archive
Python pdb for multiple processes
Python
•
Apache License 2.0
•6•36•0•0•Updated Nov 5, 2022Nov 5, 2022
engineering-class
Public
Lightning Bits: Engineering for Researchers repo
Python
•16•131•0•0•Updated Aug 29, 2022Aug 29, 2022