cattrs>=24.1.2
click>=8.1.8
einops>=0.8.0
equinox>=0.11.11
huggingface-hub[hf-transfer]>=0.27.1
jaxtyping>=0.2.36
ml-dtypes>=0.5.1
optax>=0.2.4
rich>=14.0.0
thefuzz>=0.22.1
typer>=0.15.1

[:sys_platform == "darwin"]
jax>=0.4.38

[:sys_platform == "linux"]
jax[cuda]>=0.4.38
