jaxopt (conda): Hardware accelerated, batchable and differentiable optimizers in JAX
Keywords: deep-learning|jax|optimization