requirements.txt

absl-py==2.2.2
aiofiles==25.1.0
array-record==0.8.3
astroid==3.3.9
astunparse==1.6.3
attrs==25.3.0
certifi==2025.1.31
cfgv==3.4.0
charset-normalizer==3.4.1
chex==0.1.89
clu==0.0.12
dill==0.4.0
distlib==0.3.9
dm-tree==0.1.9
docstring-parser==0.16
einops==0.8.1
etils==1.12.2
fiddle==0.3.0
filelock==3.18.0
flatbuffers==25.2.10
fsspec==2025.3.2
gast==0.6.0
google-pasta==0.2.0
googleapis-common-protos==1.70.0
graphviz==0.20.3
grpcio==1.71.0
h5py==3.13.0
humanize==4.12.2
identify==2.6.9
idna==3.10
immutabledict==4.2.1
importlib-resources==6.5.2
iniconfig==2.1.0
isort==6.0.1
jaxtyping==0.3.1
Jinja2==3.1.6
kagglehub==0.3.11
keras==3.9.2
keras-hub==0.20.0
libclang==18.1.1
libcst==1.7.0
libtpu==0.0.32
# libtpu-nightly is usually installed directly via URL, but pinning it helps tracking
# libtpu-nightly==0.1.dev20240617+default 
Markdown==3.8
markdown-it-py==3.0.0
MarkupSafe==3.0.2
mccabe==0.7.0
mdurl==0.1.2
ml-collections==1.1.0
ml-dtypes==0.5.1
mpmath==1.3.0
msgpack==1.1.0
namex==0.0.8
nest-asyncio==1.6.0
networkx==3.4.2
nodeenv==1.9.1
numpy==2.1.3
opt-einsum==3.4.0
optax==0.2.4
optree==0.15.0
orbax-checkpoint==0.11.31
packaging==24.2
platformdirs==4.3.7
pluggy==1.5.0
portpicker==1.6.0
pre-commit==4.2.0
promise==2.3
psutil==7.0.0
pyarrow==19.0.1
Pygments==2.19.1
pylint==3.3.6
pytest==8.3.5
pytest-env==1.1.5
PyYAML==6.0.2
regex==2024.11.6
requests==2.32.3
rich==14.0.0
scipy==1.15.2
setuptools==78.1.0
simple-parsing==0.1.7
simplejson==3.20.1
six==1.17.0
sympy==1.13.1
tensorboard==2.19.0
tensorboard-data-server==0.7.2
tensorflow==2.19.0
tensorflow-datasets==4.9.8
tensorflow-io-gcs-filesystem==0.37.1
tensorflow-metadata==1.17.1
tensorflow-text==2.19.0
tensorstore==0.1.80
termcolor==3.0.1
toml==0.10.2
tomlkit==0.13.2
toolz==1.0.0
torch==2.6.0
tqdm==4.67.1
treescope==0.1.9
triton==3.2.0
typing_extensions==4.13.2
urllib3==2.4.0
virtualenv==20.30.0
wadler-lindig==0.1.5
Werkzeug==3.1.3
wheel==0.45.1
wrapt==1.17.2
zipp==3.21.0