#

jit-compilation

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 Feb 25, 2025
Jupyter Notebook

Improve this page

Add a description, image, and links to the jit-compilation 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 jit-compilation topic, visit your repo's landing page and select "manage topics."