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