requirements.txt

# Direct dependencies.
absl-py==0.12.0
chex==0.0.7
dm-haiku==0.0.4
einops==0.3.0
jax==0.2.16
jaxlib==0.1.68
jaxline==0.0.3
ml-collections==0.1.0
numpy==1.19.5
optax==0.0.8
tensorflow==2.5.0
tensorflow-datasets==4.3.0
torch==1.9.0
torchvision==0.10.0
tqdm==4.61.1
# Transitive dependencies.
astunparse==1.6.3
attrs==21.2.0
cachetools==4.2.2
certifi==2021.5.30
chardet==4.0.0
contextlib2==21.6.0
dill==0.3.4
dm-tree==0.1.6
flatbuffers==1.12
future==0.18.2
gast==0.4.0
google-auth==1.32.0
google-auth-oauthlib==0.4.4
google-pasta==0.2.0
googleapis-common-protos==1.53.0
grpcio==1.34.1
h5py==3.1.0
idna==2.10
keras-nightly==2.5.0.dev2021032900
Keras-Preprocessing==1.1.2
Markdown==3.3.4
oauthlib==3.1.1
opt-einsum==3.3.0
Pillow==8.2.0
pkg-resources==0.0.0
promise==2.3
protobuf==3.17.3
pyasn1==0.4.8
pyasn1-modules==0.2.8
PyYAML==5.4.1
requests==2.25.1
requests-oauthlib==1.3.0
rsa==4.7.2
scipy==1.7.0
six==1.15.0
tabulate==0.8.9
tensorboard==2.5.0
tensorboard-data-server==0.6.1
tensorboard-plugin-wit==1.8.0
tensorflow-estimator==2.5.0
tensorflow-metadata==1.1.0
termcolor==1.1.0
toolz==0.11.1
typing-extensions==3.7.4.3
urllib3==1.26.6
Werkzeug==2.0.1
wrapt==1.12.1