PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily write your own.
☆1,473Sep 21, 2026Updated this week
Alternatives and similar repositories for lightning-thunder
Users that are interested in lightning-thunder are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Speed up model training by fixing data loading.☆615Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆404May 31, 2026Updated 3 months ago
- PyTorch native quantization for training and inference☆2,983Updated this week
- A minimal Python framework for building custom AI inference servers with full control over logic, batching, and scaling.☆3,943Updated this week
- 20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.☆13,678Sep 14, 2026Updated last week
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- PyTorch native post-training library☆5,812Sep 9, 2026Updated 2 weeks ago
- TensorDict is a pytorch dedicated tensor container.☆1,035Updated this week
- A PyTorch native platform for training generative AI models☆5,757Updated this week
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,255Aug 22, 2025Updated last year
- Freeing data processing from scripting madness by providing a set of platform-agnostic customizable pipeline processing blocks.☆3,347Updated this week
- Schedule-Free Optimization in PyTorch☆2,323Jul 28, 2026Updated last month
- Minimalistic large language model 3D-parallelism training☆2,825Updated this week
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,472Updated this week
- Save, load, host, and share AI model checkpoints without slowing down training. Host on Lightning AI or your own cloud with enterprise-gr…☆40Sep 7, 2026Updated 2 weeks ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on H…☆3,549Updated this week
- Accessible large language models via k-bit quantization for PyTorch.☆8,498Sep 7, 2026Updated 2 weeks ago
- Efficient Triton Kernels for LLM Training☆6,627Updated this week
- Development repository for the Triton language and compiler☆20,218Updated this week
- Fast and memory-efficient exact attention☆24,999Updated this week
- Tile primitives for speedy kernels☆3,724Sep 12, 2026Updated last week
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆31,355Updated this week
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,550Updated this week
- Supercharge Your Model Training☆5,499Apr 29, 2026Updated 4 months ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Tools for merging pretrained large language models.☆7,365Sep 12, 2026Updated last week
- Machine Learning Engineering Open Book☆19,031Sep 12, 2026Updated last week
- PyTorch extensions for high performance and large scale training.☆3,405Apr 26, 2025Updated last year
- Foundation Architecture for (M)LLMs☆3,141Apr 11, 2024Updated 2 years ago
- Puzzles for learning Triton☆2,607Apr 1, 2026Updated 5 months ago
- A minimal Python logger that tracks everything you try when building AI - metrics, prompts, models, etc, so you can see what changed and …☆37Updated this week
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,725Aug 7, 2026Updated last month
- Sparsity-aware deep learning inference runtime for CPUs☆3,157Jun 2, 2025Updated last year
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,604Aug 26, 2026Updated 3 weeks ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad…☆6,081Jul 1, 2025Updated last year
- A pytorch quantization backend for optimum☆1,053Aug 25, 2026Updated 3 weeks ago
- Official repository of Evolutionary Optimization of Model Merging Recipes☆1,442Nov 29, 2024Updated last year
- Placeholder for the opensource Grid AI components☆44Mar 27, 2026Updated 5 months ago
- PyTorch Single Controller☆1,076Updated this week
- LLM training in simple, raw C/CUDA☆31,046Jun 26, 2025Updated last year
- A Data Streaming Library for Efficient Neural Network Training☆1,561Jun 25, 2026Updated 2 months ago