PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily write your own.
☆1,467Jul 20, 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.☆604Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆396May 31, 2026Updated last month
- PyTorch native quantization and sparsity for training and inference☆2,912Updated this week
- A minimal Python framework for building custom AI inference servers with full control over logic, batching, and scaling.☆3,919Updated this week
- 20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.☆13,497Updated this week
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- PyTorch native post-training library☆5,785Updated this week
- TensorDict is a pytorch dedicated tensor container.☆1,034Updated this week
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,229Aug 22, 2025Updated 11 months ago
- A PyTorch native platform for training generative AI models☆5,556Updated this week
- Freeing data processing from scripting madness by providing a set of platform-agnostic customizable pipeline processing blocks.☆3,220Updated this week
- Schedule-Free Optimization in PyTorch☆2,317Jun 18, 2026Updated last month
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,448Jul 16, 2026Updated last week
- Minimalistic large language model 3D-parallelism training☆2,764May 26, 2026Updated last month
- Save, load, host, and share AI model checkpoints without slowing down training. Host on Lightning AI or your own cloud with enterprise-gr…☆40Jul 6, 2026Updated 2 weeks ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Accessible large language models via k-bit quantization for PyTorch.☆8,338Updated this week
- Efficient Triton Kernels for LLM Training☆6,533Updated this week
- Development repository for the Triton language and compiler☆19,778Updated this week
- Fast and memory-efficient exact attention☆24,526Updated 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,446Updated this week
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆31,248Updated this week
- Tile primitives for speedy kernels☆3,563Jul 13, 2026Updated last week
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,530Jul 15, 2026Updated last week
- Supercharge Your Model Training☆5,488Apr 29, 2026Updated 2 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Tools for merging pretrained large language models.☆7,260Jun 17, 2026Updated last month
- Machine Learning Engineering Open Book☆18,462Updated this week
- PyTorch extensions for high performance and large scale training.☆3,411Apr 26, 2025Updated last year
- Foundation Architecture for (M)LLMs☆3,133Apr 11, 2024Updated 2 years 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,726Jul 14, 2026Updated last week
- Sparsity-aware deep learning inference runtime for CPUs☆3,160Jun 2, 2025Updated last year
- Puzzles for learning Triton☆2,537Apr 1, 2026Updated 3 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,558Jul 5, 2026Updated 2 weeks ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad…☆6,082Jul 1, 2025Updated last year
- A pytorch quantization backend for optimum☆1,048Updated this week
- Official repository of Evolutionary Optimization of Model Merging Recipes☆1,437Nov 29, 2024Updated last year
- Placeholder for the opensource Grid AI components☆44Mar 27, 2026Updated 3 months ago
- PyTorch Single Controller☆1,060Updated this week
- LLM training in simple, raw C/CUDA☆30,631Jun 26, 2025Updated last year
- A Data Streaming Library for Efficient Neural Network Training☆1,536Jun 25, 2026Updated 3 weeks ago