evanatyourservice / psgd_jax

Implementation of PSGD optimizer in JAX
11Updated this week

Related projects

Alternatives and complementary repositories for psgd_jax