PyPI page
Author:
Evan Walters, Omead Pooladzandi, Xi-Lin Li
Summary:
An implementation of PSGD optimizer in JAX.
Latest version:
0.2.9
Required dependencies:
jax
|
matplotlib
|
numpy
|
optax
Downloads last day:
0
Downloads last week:
6
Downloads last month:
135
7-day smoothing