PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily write your own.
☆1,444Updated this week
Alternatives and similar repositories for lightning-thunder
Users that are interested in lightning-thunder are comparing it to the libraries listed below
Sorting:
- Speed up model training by fixing data loading.☆577Feb 20, 2026Updated last week
- PyTorch native quantization and sparsity for training and inference☆2,707Updated this week
- PyTorch native post-training library☆5,691Updated this week
- 20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.☆13,182Feb 22, 2026Updated last week
- A minimal Python framework for building custom AI inference servers with full control over logic, batching, and scaling.☆3,803Feb 19, 2026Updated last week
- A PyTorch native platform for training generative AI models☆5,098Updated this week
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,184Aug 22, 2025Updated 6 months ago
- TensorDict is a pytorch dedicated tensor container.☆1,009Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆379Updated this week
- Minimalistic large language model 3D-parallelism training☆2,579Feb 19, 2026Updated last week
- Freeing data processing from scripting madness by providing a set of platform-agnostic customizable pipeline processing blocks.☆2,903Updated this week
- Schedule-Free Optimization in PyTorch☆2,257May 21, 2025Updated 9 months ago
- Accessible large language models via k-bit quantization for PyTorch.☆7,997Updated this week
- Fast and memory-efficient exact attention☆22,361Updated this week
- Efficient Triton Kernels for LLM Training☆6,162Updated this week
- 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,176Updated this week
- Development repository for the Triton language and compiler☆18,501Updated this week
- Sparsity-aware deep learning inference runtime for CPUs☆3,159Jun 2, 2025Updated 8 months ago
- Tile primitives for speedy kernels☆3,183Updated this week
- Tools for merging pretrained large language models.☆6,814Jan 26, 2026Updated last month
- Foundation Architecture for (M)LLMs☆3,135Apr 11, 2024Updated last year
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,353Feb 20, 2026Updated last week
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,860Feb 21, 2026Updated last week
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,410Updated this week
- Machine Learning Engineering Open Book☆17,162Feb 21, 2026Updated last week
- Puzzles for learning Triton☆2,314Nov 18, 2024Updated last year
- A pytorch quantization backend for optimum☆1,025Nov 21, 2025Updated 3 months ago
- Official repository of Evolutionary Optimization of Model Merging Recipes☆1,399Nov 29, 2024Updated last year
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,706Jan 12, 2026Updated last month
- SGLang is a high-performance serving framework for large language models and multimodal models.☆23,905Updated this week
- Minimal, clean code for the Byte Pair Encoding (BPE) algorithm commonly used in LLM tokenization.☆10,329Jul 1, 2024Updated last year
- [ICLR 2025] Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling☆952Nov 16, 2025Updated 3 months ago
- Fast, flexible LLM inference☆6,623Updated this week
- Go ahead and axolotl questions☆11,335Updated this week
- TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizat…☆12,938Updated this week
- PyTorch extensions for high performance and large scale training.☆3,400Apr 26, 2025Updated 10 months ago
- Mamba SSM architecture☆17,257Feb 18, 2026Updated last week
- A simple, performant and scalable Jax LLM!☆2,148Updated this week
- DSPy: The framework for programming—not prompting—language models☆32,381Updated this week