GitHub repository: rezaaskary/probabilistic-modeling
Description: A probabilistic programing library empowered with jax to speed up large simulations (still in progress!)