[sito42]

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

[canyon289]

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/fas...

Between all these tools we hope the modern bayesian can find the one that works best for them