requirements.txt

absl-py==2.3.1
aiofiles==25.1.0
appnope==0.1.4
asttokens==3.0.0
attrs==25.4.0
certifi==2025.10.5
charset-normalizer==3.4.4
chex==0.1.91
cloudpickle==3.1.2
comm==0.2.3
contourpy==1.3.3
cycler==0.12.1
debugpy==1.8.17
decorator==5.2.1
distrax==0.1.7
dm-haiku==0.0.15
dm-tree==0.1.9
docstring_parser==0.17.0
einops==0.8.1
etils==1.13.0
executing==2.2.1
flax==0.12.0
fonttools==4.60.1
fsspec==2025.10.0
gast==0.6.0
googleapis-common-protos==1.72.0
hilbert-toolkit==0.1.4
humanize==4.14.0
idna==3.11
immutabledict==4.2.2
importlib_resources==6.5.2
ipykernel==7.1.0
ipython==9.7.0
ipython_pygments_lexers==1.1.1
jax==0.8.0
jaxlib==0.8.0
jaxopt==0.8.5
jedi==0.19.2
jmp==0.0.4
joblib==1.5.2
jupyter_client==8.6.3
jupyter_core==5.9.1
kfac-jax==0.0.7
kiwisolver==1.4.9
markdown-it-py==4.0.0
matplotlib==3.10.7
matplotlib-inline==0.2.1
mdurl==0.1.2
ml_dtypes==0.5.3
msgpack==1.1.2
nest-asyncio==1.6.0
numpy==2.3.4
opt_einsum==3.4.0
optax==0.2.6
orbax-checkpoint==0.11.28
packaging==25.0
parso==0.8.5
pexpect==4.9.0
pillow==12.0.0
platformdirs==4.5.0
promise==2.3
prompt_toolkit==3.0.52
protobuf==6.33.0
psutil==7.1.3
ptyprocess==0.7.0
pure_eval==0.2.3
pyarrow==22.0.0
Pygments==2.19.2
pyparsing==3.2.5
python-dateutil==2.9.0.post0
PyYAML==6.0.3
pyzmq==27.1.0
requests==2.32.5
rich==14.2.0
scikit-learn==1.7.2
scipy==1.16.3
simple-parsing==0.1.7
simplejson==3.20.2
six==1.17.0
stack-data==0.6.3
tabulate==0.9.0
tensorflow-datasets==4.9.9
tensorflow-metadata==1.17.2
tensorstore==0.1.78
termcolor==3.2.0
tfp-nightly==0.26.0.dev20251110
threadpoolctl==3.6.0
toml==0.10.2
toolz==1.1.0
tornado==6.5.2
tqdm==4.67.1
traitlets==5.14.3
treescope==0.1.10
typing_extensions==4.15.0
urllib3==2.5.0
wcwidth==0.2.14
wrapt==2.0.1
zipp==3.23.0