GitHub repository: jvlmdr/jaxopt
Primary language: Python
Description: Hardware accelerated, batchable and differentiable optimizers in JAX.