jax==0.8.0
jaxlib==0.8.0
flax==0.12.0
optax==0.2.6
chex==0.1.91
grain==0.2.11
tensorflow-datasets==4.9.9+nightly
pydantic==2.11.7
albumentations==2.0.8
numpy==2.3.3
matplotlib==3.10.5
pandas==2.3.1
