flax==0.10.4
safetensors
pydantic
huggingface-hub
soundfile
descript-audio-codec
torchaudio

[:platform_system == "Darwin"]
jax-metal==0.1.1
jaxlib==0.5.0
jax==0.5.0

[gpu]
jax[cuda12]

[tpu]
jax[tpu]
