tfp-nightly==0.15.0.dev20211020
jax==0.2.24
jaxlib==0.1.73
flax==0.3.5
chex==0.0.8
pyyaml==6.0
optax==0.0.9
ml-collections==0.1.0
gym==0.21.0
tensorboardX==2.4.0
tqdm==4.62.3
tensorflow==2.7.0rc1
numpy>=1.21.3
