{"library":"heavyball","type":"library","category":null,"description":"HeavyBall is a PyTorch optimizer library that emphasizes 'compile-first' design, assembling optimizers from composable, compiled building blocks. It provides API-compatible replacements for `torch.optim` optimizers like AdamW, SGD, and RMSprop, along with over 30 specialized optimizers such as Muon, SOAP/Shampoo, PSGD, and Schedule-Free. Currently at version 3.0.0, the library is actively maintained with a focus on `torch.compile` fusion, Triton kernel optimization, and memory efficiency, including features like ECC state compression.","language":"python","status":"active","version":"3.0.0","tags":["pytorch","optimizer","deep-learning","machine-learning","torch-compile","triton","mixed-precision","memory-efficient","adamw","sgd","soap","psgd","muon"],"last_verified":"Tue May 26","install":[{"cmd":"pip install heavyball","imports":["from heavyball import AdamW","from heavyball import SOAP","from heavyball import Muon"]}],"homepage":null,"github":"https://github.com/HomebrewML/HeavyBall","docs":null,"changelog":null,"pypi":"https://pypi.org/project/heavyball/","npm":null,"openapi_spec":null,"status_page":null,"smithery":null,"compatibility":{"summary":{"python_range":"3.10–3.9","success_rate":90,"avg_install_s":65.6,"avg_import_s":11.6,"wheel_type":"wheel"},"url":"https://checklist.day/v1/registry/heavyball/compatibility"}}