evanatyourservice / psgd_jax

Implementation of PSGD optimizer in JAX
19Updated 2 weeks ago

Related projects

Alternatives and complementary repositories for psgd_jax