requirements.txt

absl-py==1.4.0
antlr4-python3-runtime==4.9.3
array-record==0.4.0
astunparse==1.6.3
cachetools==5.3.1
certifi==2023.5.7
charset-normalizer==3.1.0
click==8.1.3
colorlog==6.7.0
dm-tree==0.1.8
einops==0.6.1
etils==1.3.0
flatbuffers==23.5.26
gast==0.4.0
gin-config==0.5.0
google-auth==2.21.0
google-auth-oauthlib==1.0.0
google-pasta==0.2.0
googleapis-common-protos==1.59.1
grpcio==1.56.0
h5py==3.9.0
hydra-colorlog==1.2.0
hydra-core==1.3.2
idna==3.4
importlib-resources==5.12.0
jax==0.4.13
jaxlib==0.4.13+cuda12.cudnn89
keras==2.12.0
libclang==16.0.0
Markdown==3.4.3
MarkupSafe==2.1.3
ml-dtypes==0.2.0
numpy==1.23.5
nvidia-cublas-cu12==12.2.1.16
nvidia-cuda-cupti-cu12==12.2.60
nvidia-cuda-nvcc-cu12==12.2.91
nvidia-cuda-runtime-cu12==12.2.53
nvidia-cudnn-cu12==8.9.2.26
nvidia-cufft-cu12==11.0.8.15
nvidia-cusolver-cu12==11.5.0.53
nvidia-cusparse-cu12==12.1.1.53
nvidia-nvjitlink-cu12==12.2.91
oauthlib==3.2.2
omegaconf==2.3.0
opt-einsum==3.3.0
packaging==23.1
promise==2.3
protobuf==4.23.3
psutil==5.9.5
pyasn1==0.5.0
pyasn1-modules==0.3.0
PyYAML==6.0
requests==2.31.0
requests-oauthlib==1.3.1
rsa==4.9
scipy==1.11.1
six==1.16.0
tensorboard==2.12.3
tensorboard-data-server==0.7.1
tensorboardX==2.6.1
tensorflow==2.12.0
tensorflow-datasets==4.9.2
tensorflow-estimator==2.12.0
tensorflow-io-gcs-filesystem==0.32.0
tensorflow-metadata==1.13.1
termcolor==2.3.0
toml==0.10.2
tqdm==4.65.0
typing_extensions==4.7.1
urllib3==1.26.16
Werkzeug==2.3.6
wrapt==1.14.1
zipp==3.15.0