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