tensorflow_probability
jaxlib
jax
numpy
setuptools_scm[toml]

setuptools_scm_git_archive
