cattrs[msgpack]>=24.1.2
click>=8.1.8
einops>=0.8.0
equinox>=0.11.11
huggingface-hub[hf-transfer]>=0.27.1
jax>=0.8.1
jaxtyping>=0.2.36
jinja2>=3.1.6
ml-dtypes>=0.5.1
optax>=0.2.4
rich>=14.0.0
thefuzz>=0.22.1
tokenizers>=0.21.2
typer>=0.15.1
polars>=1.33.1
xxhash>=3.5.0

[cpu]
jax[cpu]>=0.8.1

[cuda]
jax[cuda]>=0.8.1

[tpu]
jax[tpu]>=0.8.1
