JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro
语言:Jupyter Notebook · GitHub 星标:19 · Fork:2
话题:autodifferentiation · jax · jaxoptim · jit-compilation · numpy · numpyro · scipy · tutorial-demos
在 GitHub 上查看原仓库