jecampagne/JaxTutos

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

Language: Jupyter Notebook - Size: 35.6 MB - Last synced at: about 1 month ago - Pushed at: 3 months ago - Stars: 14 - Forks: 2