absl-py==2.3.1
antlr4-python3-runtime==4.9.3
appdirs==1.4.4
blackjax==1.2.5
certifi==2025.6.15
charset-normalizer==3.4.2
chex==0.1.85
click==8.2.1
contourpy==1.3.2
cycler==0.12.1
docker-pycreds==0.4.0
einops==0.7.0
et_xmlfile==2.0.0
execnet==2.1.1
fastprogress==1.0.3
flax==0.8.5
fonttools==4.58.5
gitdb==4.0.12
GitPython==3.1.44
hydra-core==1.3.2
idna==3.10
iniconfig==2.1.0
jax==0.6.2
jaxlib==0.6.2
jaxopt==0.8.5
jaxtyping==0.2.25
kiwisolver==1.4.8
markdown-it-py==3.0.0
matplotlib==3.9.2
mdurl==0.1.2
memory-profiler==0.61.0
ml_dtypes==0.5.1
mypy==1.16.1
mypy_extensions==1.1.0
numpy>=2.1.0
omegaconf==2.3.0
openpyxl==3.1.5
opt_einsum==3.4.0
optax==0.1.9
orbax-checkpoint==0.4.4
ott-jax==0.4.5
packaging==25.0
pandas==2.3.0
pathspec==0.12.1
pillow==11.3.0
plotly==5.17.0
pluggy==1.6.0
protobuf==4.25.8
psutil==5.9.6
Pygments==2.19.2
pyparsing==3.2.3
pytest==8.4.1
pytest-xdist==3.8.0
optuna==3.6.1
python-dateutil==2.9.0.post0
pytz==2025.2
PyYAML==6.0.2
requests==2.32.4
rich==13.7.0
scipy==1.16.1
seaborn==0.13.0
sentry-sdk==2.32.0
setproctitle==1.3.6
six==1.17.0
smmap==5.0.2
tenacity==9.1.2
toolz==1.0.0
typeguard==2.13.3
typing_extensions==4.14.1
tzdata==2025.2
urllib3==2.5.0
wandb==0.16.1
