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:
3,096
Downloads last week:
10,968
Downloads last month:
69,732