#

jaxoptim

Here is 1 public repository matching this topic...

jecampagne / JaxTutos

JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro

numpy scipy jax autodifferentiation tutorial-demos jit-compilation numpyro jaxoptim

Updated Nov 16, 2024
Jupyter Notebook

Improve this page

Add a description, image, and links to the jaxoptim topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the jaxoptim topic, visit your repo's landing page and select "manage topics."