Lightning-AI / lightning-thunder

git clone https://github.com/Lightning-AI/lightning-thunder.git

PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily write your own.