array_record==0.5.1
wandb==0.18.7
nvidia-curand-cu12==10.3.2.106
requests-oauthlib==2.0.0
zipp==3.21.0
Werkzeug==3.1.3
simple-parsing==0.1.6
mdurl==0.1.2
keras==2.15.0
nvidia-cuda-nvcc-cu12==12.6.85
google-auth-oauthlib==1.2.1
jaxlib==0.4.34
tf_keras==2.15.1
oauthlib==3.2.2
tensorflow-probability==0.25.0
cachetools==5.5.0
Jinja2==3.1.3
rich==13.9.4
filelock==3.13.1
google-pasta==0.2.0
optax==0.2.4
toolz==1.0.0
gast==0.6.0
tensorboard==2.15.2
pyasn1_modules==0.4.1
nvidia-cudnn-cu12==9.1.0.70
opt_einsum==3.4.0
nvidia-nvjitlink-cu12==12.6.85
chex==0.1.87
namex==0.0.8
termcolor==2.5.0
flax==0.10.2
cloudpickle==3.1.0
numpy==1.26.4
nvidia-nccl-cu12==2.21.5
tensorflow-cpu==2.15.0
nvidia-cusolver-cu12==11.4.5.107
typing_extensions==4.12.2
tensorflow-addons==0.23.0
typeguard==2.13.3
absl-py==2.1.0
flatbuffers==24.3.25
dlpack==0.1
setuptools==65.5.0
protobuf==4.25.5
jax-cuda12-plugin==0.4.35
tensorflow==2.15.0
msgpack==1.1.0
networkx==3.2.1
docker-pycreds==0.4.0
nvidia-cuda-cupti-cu12==12.1.105
pillow==11.0.0
libclang==18.1.1
nvidia-cuda-nvrtc-cu12==12.1.105
distrax==0.1.5
orbax-checkpoint==0.10.1
PyYAML==6.0.2
urllib3==2.2.3
aqtp==0.8.2
tensorflow-metadata==1.16.1
etils==1.11.0
smmap==5.0.1
pyasn1==0.6.1
docstring_parser==0.16
google-auth==2.36.0
simplejson==3.19.3
mpmath==1.3.0
h5py==3.12.1
jax-cuda12-pjrt==0.4.35
tensorflow-io-gcs-filesystem==0.37.1
tensorflow-estimator==2.15.0
pydlpack==0.2.1
triton==3.1.0
rsa==4.9
panopticapi==0.1
tensorflow-hub==0.16.1
requests==2.32.3
scipy==1.14.1
ml-dtypes==0.2.0
markdown-it-py==3.0.0
tensorflow-text==2.15.0
wrapt==1.14.1
immutabledict==4.2.1
MarkupSafe==3.0.2
jax==0.4.35
torch==2.5.1+cu121
wheel==0.45.1
einops==0.8.0
sentry-sdk==2.19.0
torchvision==0.20.1+cu121
humanize==4.11.0
toml==0.10.2
tensorstore==0.1.69
six==1.16.0
promise==2.3
certifi==2024.8.30
nvidia-cuda-runtime-cu12==12.1.105
flaxformer==0.8.8
nvidia-cufft-cu12==11.0.2.54
psutil==6.1.0
GitPython==3.1.43
platformdirs==4.3.6
importlib_resources==6.4.5
tfds-nightly==4.9.7.dev202411280044
tensorflow-gan==2.1.0
googleapis-common-protos==1.66.0
overrides==7.7.0
optree==0.13.1
Pygments==2.18.0
astunparse==1.6.3
ml_collections==1.0.0
setproctitle==1.3.4
tensorboard-data-server==0.7.2
sympy==1.13.1
packaging==24.2
nest-asyncio==1.6.0
nvidia-cublas-cu12==12.1.3.1
gitdb==4.0.11
click==8.1.7
idna==3.10
tqdm==4.67.1
grpcio==1.68.0
decorator==5.1.1
pyarrow==18.1.0
clu==0.0.12
charset-normalizer==3.4.0
fsspec==2024.10.0
dm-tree==0.1.8
sentencepiece==0.2.0
nvidia-cusparse-cu12==12.1.0.106
torchaudio==2.5.1+cu121
pip==24.3.1
Markdown==3.7
nvidia-nvtx-cu12==12.1.105