numpyroExpert Playground

Bayesian modeling on JAX for GPU-accelerated MCMC sampling

numpyro expert patternsRun locally
Install
pip install numpyro
Python CodeRun locally

Expert-level numpyro usage for performance-critical and production-grade applications.

Challenge

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