Thunder – A new compiler for PyTorch that can use many device executors at once(github.com)

11 pointsby wfalconhace 2 años1 comments

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

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

1 comments

rasbthace 2 años

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.