PyPI page
Home page
Author:
HeavyBall Authors
Summary:
Compile-first PyTorch optimizer library - AdamW, Muon, SOAP/Shampoo, PSGD, Schedule-Free, and 30+ more with torch.compile fusion and composable features
Latest version:
3.2.1
Required dependencies:
numpy
|
opt-einsum
|
torch
Optional dependencies:
gpytorch
|
hypothesis
|
matplotlib
|
optuna
|
optunahub
|
pandas
|
pre-commit
|
pytest
|
ruff
|
scikit-learn
|
seaborn
|
tqdm
|
typer
Downloads last day:
1,032
Downloads last week:
7,254
Downloads last month:
30,163
7-day smoothing