requirements.txt

absl-py==1.4.0
aiohttp>=3.8.5
aiosignal==1.3.1
async-timeout==4.0.2
attrs==23.1.0
cached-property==1.5.2
cachetools==5.3.1
certifi>=2023.07.22
cfgv==3.3.1
charset-normalizer==2.1.1
chex==0.1.81
contextlib2==21.6.0
contourpy==1.1.0
cycler==0.11.0
distlib==0.3.6
dm-tree==0.1.8
etils==1.3.0
filelock==3.12.2
flax==0.7.0
fonttools>=4.43.0
frozenlist==1.3.3
fsspec==2023.6.0
google-auth==2.21.0
google-auth-oauthlib==1.0.0
grpcio==1.56.0
identify==2.5.24
idna==3.4
importlib-resources==6.0.0
jax==0.4.13
jaxlib==0.4.13+cuda12.cudnn89
Jinja2>=3.1.3
kiwisolver==1.4.4
lightning-utilities==0.9.0
Markdown==3.4.3
markdown-it-py==3.0.0
MarkupSafe==2.1.2
matplotlib==3.7.2
mdurl==0.1.2
ml-collections==0.1.1
ml-dtypes==0.2.0
mpmath>=1.3.0
msgpack==1.0.5
multidict==6.0.4
nest-asyncio==1.5.6
networkx==3.0
nodeenv==1.8.0
numpy==1.25.1
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
opt-einsum==3.3.0
optax==0.1.5
orbax-checkpoint==0.2.7
packaging==23.1
pandas==2.0.3
Pillow>=10.0.1
platformdirs==3.8.1
pre-commit==3.3.3
protobuf==4.23.4
pyasn1==0.5.0
pyasn1-modules==0.3.0
Pygments==2.15.1
pyparsing==3.0.9
python-dateutil==2.8.2
pytorch-lightning==2.0.5
pytz==2023.3
PyYAML==6.0
requests>=2.31.0
requests-oauthlib>=1.3.1
rich==13.4.2
rsa==4.9
scipy==1.11.1
seaborn==0.12.2
six==1.16.0
sympy==1.11.1
tabulate==0.9.0
tensorboard==2.13.0
tensorboard-data-server==0.7.1
tensorstore==0.1.40
toolz==0.12.0
torch==2.0.1+cpu
torchaudio==2.0.2+cpu
torchmetrics==1.0.0
torchvision==0.15.2+cpu
tqdm==4.65.0
typing_extensions==4.4.0
tzdata==2023.3
urllib3>=1.26.18
virtualenv==20.23.1
Werkzeug>=2.3.8
yarl==1.9.2
zipp==3.16.0