JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro
-
Updated
Feb 25, 2025 - Jupyter Notebook
#
numpyro
Here are 4 public repositories matching this topic...
Summary notebook implementing Bayesian Model Averaging with numpyro.
-
Updated
May 20, 2024 - Jupyter Notebook
Summary notebooks using derivative gaussian processes with tinygp. We implement a 2D derivative gaussian process and successfully use derivatives to regularize SVI fits with a gaussian process model..
-
Updated
May 16, 2024 - Jupyter Notebook
Improve this page
Add a description, image, and links to the numpyro topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the numpyro topic, visit your repo's landing page and select "manage topics."