absl-py
numpy
#jax
#jaxlib
pytest
tqdm
flax
dm-haiku
optax
tensorflow
tensorflow-datasets
tensorflow-metadata
tensorflow-probability
tensorboard
gin-config
seqio
oryx
