absl-py==0.12.0
numpy>=1.18
jax>=0.2.6
jaxlib>=0.1.68
nose
dm-launchpad-nightly==0.3.0.dev20211105
tqdm>=4.62.3
flax==0.3.3
dm-haiku==0.0.5
optax>=0.0.9
tensorflow>=2.7.0
tensorflow-datasets>=4.4.0
tensorflow-metadata==1.5.0
tensorboard>=2.7.0
gin-config>=0.5.0
