matplotlib~=3.10.3
tqdm~=4.67.1
tokenizers~=0.21.2
safetensors~=0.5.3
mlflow~=3.1.1
mlflow-skinny~=3.1.1
numpy~=2.3.1
jax~=0.7.1
flax~=0.11.2
polars~=1.32.3
optax~=0.2.5