Honestly, I'm not familiar with it. I had only played with RxInfer, if only to try the so-called "message-passing" paradigm. My grasp on probability is really lacking, in fact I picked up prediction markets and Julia to get better at it.
If you don't mind me asking, what's the deal with NumPyro why you chose to work it?
Well I did my PhD working with Probabilistic Torch, then shifted to Pyro for most applied probabilistic programming in PyTorch, and then wanted to learn Jax. So I moved over to Numpyro.
If you don't mind me asking, what's the deal with NumPyro why you chose to work it?