absl-py
six>=1.10.0
numpy>=1.13.3
decorator
cloudpickle>=1.3
gast>=0.3.2
dm-tree

[jax]
jax
jaxlib

[tfds]
tfds-nightly
