Hardware accelerated, batchable and differentiable optimizers in JAX.
-
Updated
Sep 17, 2024 - Python
Hardware accelerated, batchable and differentiable optimizers in JAX.
Add a description, image, and links to the bi-level topic page so that developers can more easily learn about it.
To associate your repository with the bi-level topic, visit your repo's landing page and select "manage topics."