โš ๏ธ Heads up: this tool's official website appears to be unreachable right now. It may be offline, rebranded, or temporarily down.
Lightning Thunder

Lightning Thunder

โ˜… 3.8 ๐Ÿ”– 1200 saves ๐Ÿ’ฐ Free GitHub

Lightning Thunder is a source-to-source compiler for PyTorch, developed by Lightning AI, that accelerates model training and inference by fusing and optimizing operations before execution. The open-source project has around 1,500 stars on GitHub and is under active development in 2026.

Overview

Thunder takes ordinary PyTorch code and compiles it into optimized programs, combining operations to reduce memory traffic and kernel-launch overhead. It works by tracing your program and distributing work across a set of executors, which can include cuDNN and cuBLAS for dense math, Triton for generated kernels, and PyTorch eager execution itself, and it composes with torch.compile rather than replacing your whole stack. A distinctive design goal is readability: Thunder exposes the generated code and intermediate representations so performance engineers can inspect exactly what will run, understand why it is fast or slow, and write their own optimizations and transforms. That makes it both a practical accelerator for common models and a research platform for compiler experiments. Released under the Apache-2.0 license, Thunder targets training and inference workloads where squeezing more throughput or lower memory use out of the same GPUs has direct cost impact.

Key Features

  • Source-to-source compilation of PyTorch programs for faster training and inference
  • Operation fusion that cuts memory traffic and kernel launch overhead
  • Multiple executors: cuDNN, cuBLAS, Triton, and eager PyTorch
  • Composes with torch.compile instead of forcing an all-or-nothing switch
  • Inspectable generated code and IR for debugging and custom transforms
  • Built-in optimizations for performance, memory use, and parallelism

๐Ÿ’ณ Pricing

Free. Lightning Thunder is open source under the Apache-2.0 license; there is no paid tier for the compiler itself.

Who Is It For

PyTorch engineers and performance researchers who need more speed or lower memory from existing models and want a compiler they can inspect and extend.

๐Ÿš€ Visit Lightning Thunder Official Site

๐Ÿšฉ Report an issue