GitHub repository: pipme/jaxopt
Description: Hardware accelerated, batchable and differentiable optimizers in JAX.