PyPI page
Home page
Author:
None
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.1.1
Required dependencies:
numpy
|
opt-einsum
|
torch
Optional dependencies:
gpytorch
|
hypothesis
|
lightbench
|
matplotlib
|
optuna
|
optunahub
|
pandas
|
pre-commit
|
pytest
|
ruff
|
scikit-learn
|
seaborn
|
typer
Downloads last day:
4,445
Downloads last week:
24,039
Downloads last month:
177,521