numpyroEasy Playground

Bayesian modeling on JAX for GPU-accelerated MCMC sampling

Getting started with numpyroRun locally
Install
pip install numpyro
Python CodeRun locally
Expected Output
# Expected output shown below
# (Run locally with: numpyro)

numpyro is a third-party package. Bayesian modeling on JAX for GPU-accelerated MCMC sampling. Install with: pip install numpyro

Challenge

Try modifying the code above to explore different behaviors. Can you extend the example to handle a new use case?