Compile-first PyTorch optimizer library - AdamW, Muon, SOAP/Shampoo, PSGD, Schedule-Free, and 30+ more with torch.compile fusion and composable features
No known vulnerabilities in the latest version
Based on latest version 3.0.0. If you're running an older version, check OSV for your specific version.
[](/packages/heavyball)
<a href="/packages/heavyball"><img src="/api/badges/heavyball?period=month" alt="PyPI Stats"></a>