Hardware accelerated, batchable and differentiable optimizers in JAX.
-
Updated
Jun 4, 2026 - Python
Hardware accelerated, batchable and differentiable optimizers in JAX.
To associate your repository with the bi-level topic, visit your repo's landing page and select "manage topics."