psgd-jax
PyPI page
Home 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:
11
Downloads last week:
45
Downloads last month:
410