jax==0.2.5
jaxlib==0.1.56
tfp-nightly==0.12.0.dev20201107

[:python_version < "3.7"]
dataclasses
