fehiepsi / rethinking-numpyro

Statistical Rethinking (2nd ed.) with NumPyro

https://fehiepsi.github.io/rethinking-numpyro/

Updates for numpyro 0.5

fehiepsi opened this issue 3 years ago · comments

Du Phan commented 3 years ago

Use svi.run instead of init+update+lax.scan+get_params.