absl-py==2.3.0
array_record==0.7.2
astunparse==1.6.3
attrs==25.3.0
certifi==2025.6.15
charset-normalizer==3.4.2
chex==0.1.89
dm-haiku==0.0.14
dm-tree==0.1.9
docstring_parser==0.16
einops==0.8.1
etils==1.12.2
flatbuffers==25.2.10
fsspec==2025.5.1
gast==0.6.0
google-pasta==0.2.0
googleapis-common-protos==1.70.0
grpcio==1.73.1
h5py==3.14.0
idna==3.10
immutabledict==4.2.1
importlib_resources==6.5.2
jax==0.4.30
jax-cuda12-pjrt==0.4.30
jax-cuda12-plugin==0.4.30
jaxlib==0.4.30
jmp==0.0.4
keras==3.10.0
libclang==18.1.1
Markdown==3.8.2
markdown-it-py==3.0.0
MarkupSafe==3.0.2
mdurl==0.1.2
ml_collections==1.1.0
ml_dtypes==0.5.1
namex==0.1.0
numpy==2.1.3
nvidia-cublas-cu12==12.5.3.2
nvidia-cuda-cupti-cu12==12.5.82
nvidia-cuda-nvcc-cu12==12.5.82
nvidia-cuda-nvrtc-cu12==12.5.82
nvidia-cuda-runtime-cu12==12.5.82
nvidia-cudnn-cu12==9.3.0.75
nvidia-cufft-cu12==11.2.3.61
nvidia-curand-cu12==10.3.6.82
nvidia-cusolver-cu12==11.6.3.83
nvidia-cusparse-cu12==12.5.1.3
nvidia-nccl-cu12==2.23.4
nvidia-nvjitlink-cu12==12.5.82
opt_einsum==3.4.0
optax==0.2.5
optree==0.16.0
packaging==25.0
promise==2.3
protobuf==5.29.5
psutil==7.0.0
pyarrow==20.0.0
Pygments==2.19.2
PyYAML==6.0.2
requests==2.32.4
rich==14.0.0
scipy==1.16.0
setuptools==80.9.0
simple-parsing==0.1.7
six==1.17.0
tabulate==0.9.0
tensorboard==2.19.0
tensorboard-data-server==0.7.2
tensorflow==2.19.0
tensorflow-metadata==1.17.2
termcolor==3.1.0
tfds-nightly==4.9.9.dev202507020044
toml==0.10.2
toolz==1.0.0
tqdm==4.67.1
typing_extensions==4.14.0
urllib3==2.5.0
Werkzeug==3.1.3
wheel==0.45.1
wrapt==1.17.2
zipp==3.23.0
