numpyro — Intermediate Playground
Bayesian modeling on JAX for GPU-accelerated MCMC sampling
numpyro intermediate patternsRun locally
Install
pip install numpyroPython CodeRun locally
These patterns demonstrate how numpyro is used in production applications.
Challenge
Try modifying the code above to explore different behaviors. Can you extend the example to handle a new use case?