jaxopt (pypi): Hardware accelerated, batchable and differentiable optimizers in JAX.
Keywords: optimization|root finding|implicit differentiation|jax|bi-level|deep-learning|differentiable-programming