Lightning-AI / lightning-thunder

Thunder gives you PyTorch models superpowers for training and inference. Unlock out-of-the-box optimizations for performance, memory and parallelism, or roll out your own.
1,314Updated this week

Alternatives and similar repositories for lightning-thunder:

Users that are interested in lightning-thunder are comparing it to the libraries listed below