PyPI page
Home page
Author:
Evan Walters, Omead Pooladzandi, Xi-Lin Li
Summary:
An implementation of PSGD Kron optimizer in PyTorch.
Latest version:
0.3.3
Required dependencies:
opt-einsum
|
torch
|
triton
Downloads last day:
92
Downloads last week:
761
Downloads last month:
2,530