3 ms·

numpyro is an underrated library that runs on jax which makes it easy to put on the gpu. They have a nice suite of examples. https://github.com/pyro-ppl/numpyr

by sito42 3y ago

numpyro is an underrated library that runs on jax which makes it easy to put on the gpu. They have a nice suite of examples. https://github.com/pyro-ppl/numpyro https://github.com/pyro-ppl/numpyro

canyon289 3y ago
Numpyro is fantastic, great syntax and great contributors as well. PyMC now supports jax and numba as well, and can use the numpyro samplers. https://www.pymc.io/projects/examples/en/latest/samplers/fast_sampling_with_jax_and_numba.html https://www.pymc.io/projects/examples/en/latest/samplers/fas... Between all these tools we hope the modern bayesian can find the one that works best for them