numpy>=1.12
jax
jaxlib
msgpack
sklearn
tensorflow
tensorflow-text
tfds-nightly
tqdm
