Skip to content

Thunder – A new compiler for PyTorch that can use many device executors at once

github.com/Lightning-AI
11 pointswfalcon1 comment
On HN

Comments

Quoting from the readme, it embraces other executers, including torch.compile and also works with multiple GPUs:

Thunder is a source-to-source compiler for PyTorch. It makes PyTorch programs faster by combining and using different hardware executors at once (ie: nvFuser, torch.compile, cuDNN, and TransformerEngine FP8).

Works on single accelerators and in multi-GPU settings. Thunder aims to be usable, understandable, and extensible.

AboutSource Built by g1lg1l

Hackerly is an independent reader for Hacker News, built on the public HN API. Not affiliated with Y Combinator.