haliax>=1.4.dev324
equinox>=0.11.7
jaxtyping>=0.2.34
tokenizers>=0.15.2
transformers<4.46.0,>=4.41.2
optax>=0.1.9
wandb>=0.17.8
draccus>=0.9.3
pyarrow>=11.0.0
zstandard>=0.20.0
datasets<4.0,>=3.1.0
gcsfs<2024.10,>=2024.2
braceexpand>=0.1.7
jmp>=0.0.3
fsspec[http]<2024.10,>=2024.2
tensorstore>=0.1.65
pytimeparse>=1.1.8
humanfriendly==10.0
safetensors[numpy]~=0.4.2
matplotlib>=3.7.0
tblib<4.0.0,>=1.7.0
dataclasses-json~=0.6.4
ray[default]>=2.34.0
pydantic<3
rich~=13.0
filelock~=3.13
async-lru~=2.0
tqdm-loggable>=0.2
deepdiff

[test]
pytest
flake8
soundfile
librosa
pytest-forked
pytest-asyncio
