tensorflow-probability
numpyro
jax

[test]
pytest>=3.6
