https://github.com/jecampagne/jaxtutos

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

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

Last synced: 11 Apr 2025