beartype>=0.22.9
cattrs[msgpack]>=24.1.2
click>=8.1.8
einops>=0.8.0
equinox>=0.11.11
huggingface-hub>=0.27.1
jax>=0.10.0
jaxtyping>=0.2.36
jinja2>=3.1.6
ml-dtypes>=0.5.1
optax>=0.2.4
orbax-checkpoint==0.11.33
polars>=1.33.1
requests>=2.32.5
rich>=14.0.0
soundfile>=0.13.1
thefuzz>=0.22.1
tokamax>=0.0.12
tokenizers>=0.21.2
typer<0.26,>=0.15.1
xxhash>=3.5.0

[cpu]
jax[cpu]>=0.10.0

[cuda]
jax[cuda]>=0.10.0

[cuda:platform_system == "Linux"]
nvidia-cutlass-dsl>=4.4

[fishaudio]
torch>=2.8.0
transformers>=5.5.0
tiktoken>=0.12.0
blobfile>=3.1.0

[pyaudio]
pyaudio>=0.2.0

[server]
fastapi>=0.135.3
uvicorn>=0.44.0

[tpu]
jax[tpu]>=0.10.0
