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.