absl-py==2.1.0
# Editable Git install with no remote (algorithmic_efficiency==0.1.0)
-e /algorithmic-efficiency
array_record==0.5.1
astunparse==1.6.3
certifi==2024.8.30
charset-normalizer==3.4.0
chex==0.1.87
click==8.1.7
clu==0.0.12
colorama==0.4.6
contourpy==1.3.1
cycler==0.12.1
dm-tree==0.1.8
docker==7.1.0
docker-pycreds==0.4.0
docstring_parser==0.16
etils==1.11.0
filelock==3.16.1
flatbuffers==24.3.25
flax==0.10.1
fonttools==4.55.2
fsspec==2024.10.0
gast==0.6.0
gitdb==4.0.11
GitPython==3.1.43
google-pasta==0.2.0
googleapis-common-protos==1.66.0
GPUtil==1.4.0
grpcio==1.68.1
h5py==3.12.1
humanize==4.11.0
idna==3.10
imageio==2.36.1
immutabledict==4.2.1
importlib_resources==6.4.5
jax==0.4.35
jax-cuda12-pjrt==0.4.35
jax-cuda12-plugin==0.4.35
jaxlib==0.4.35
Jinja2==3.1.4
joblib==1.4.2
jraph==0.0.6.dev0
keras==3.7.0
kiwisolver==1.4.7
lazy_loader==0.4
libclang==18.1.1
lxml==5.3.0
Markdown==3.7
markdown-it-py==3.0.0
MarkupSafe==3.0.2
matplotlib==3.9.3
mdurl==0.1.2
ml-dtypes==0.4.1
ml_collections==1.0.0
mpmath==1.3.0
msgpack==1.1.0
namex==0.0.8
nest-asyncio==1.6.0
networkx==3.2.1
numpy==2.0.2
nvidia-cublas-cu12==12.4.5.8
nvidia-cuda-cupti-cu12==12.4.127
nvidia-cuda-nvcc-cu12==12.6.85
nvidia-cuda-nvrtc-cu12==12.4.127
nvidia-cuda-runtime-cu12==12.4.127
nvidia-cudnn-cu12==9.1.0.70
nvidia-cufft-cu12==11.2.1.3
nvidia-curand-cu12==10.3.5.147
nvidia-cusolver-cu12==11.6.1.9
nvidia-cusparse-cu12==12.3.1.170
nvidia-nccl-cu12==2.21.5
nvidia-nvjitlink-cu12==12.4.127
nvidia-nvtx-cu12==12.4.127
opt_einsum==3.4.0
optax==0.2.4
optree==0.13.1
orbax-checkpoint==0.10.2
packaging==24.2
pandas==2.2.3
pillow==11.0.0
platformdirs==4.3.6
portalocker==3.0.0
promise==2.3
protobuf==4.25.5
psutil==6.1.0
pyarrow==18.1.0
pydub==0.25.1
Pygments==2.18.0
pyparsing==3.2.0
python-dateutil==2.9.0.post0
pytz==2024.2
PyYAML==6.0.2
regex==2024.11.6
requests==2.32.3
rich==13.9.4
sacrebleu==1.3.1
scikit-image==0.24.0
scikit-learn==1.5.2
scipy==1.14.1
sentencepiece==0.2.0
sentry-sdk==2.19.2
setproctitle==1.3.4
simple-parsing==0.1.6
simplejson==3.19.3
six==1.17.0
smmap==5.0.1
sympy==1.13.1
tabulate==0.9.0
tensorboard==2.18.0
tensorboard-data-server==0.7.2
tensorflow==2.18.0
tensorflow-addons==0.23.0
tensorflow-datasets==4.9.7
tensorflow-io-gcs-filesystem==0.37.1
tensorflow-metadata==1.16.1
tensorflow-text==2.18.0
tensorstore==0.1.69
termcolor==2.5.0
threadpoolctl==3.5.0
tifffile==2024.9.20
toml==0.10.2
toolz==1.0.0
torch==2.5.1
torchvision==0.20.1
tqdm==4.67.1
triton==3.1.0
typeguard==2.13.3
typing==3.7.4.3
typing_extensions==4.12.2
tzdata==2024.2
urllib3==2.2.3
wandb==0.18.7
Werkzeug==3.1.3
wrapt==1.17.0
zipp==3.21.0