tensorflow_probability
jaxlib
jax
numpy
setuptools_scm[toml]
setuptools_scm_git_archive
