PyPI page
Home page
Author:
None
Summary:
A library for unit scaling in PyTorch, based on the paper 'u-muP: The Unit-Scaled Maximal Update Parametrization.'
Latest version:
0.3.5
Required dependencies:
docstring-parser
|
einops
|
tabulate
|
torch
Optional dependencies:
check-manifest
|
datasets
|
matplotlib
|
pandas
|
pytest
|
seaborn
Downloads last day:
408
Downloads last week:
2,595
Downloads last month:
38,107