haliax>=1.4.dev307
equinox>=0.11.4
jaxtyping>=0.2.20
tokenizers>=0.15.2
transformers>=4.41.2
optax>=0.1.9
wandb<0.17.6,>=0.16.6
scipy<=1.12.0
draccus>=0.8.0
pyarrow>=11.0.0
zstandard>=0.20.0
datasets~=2.18
gcsfs<2024.7,>=2024.2
braceexpand>=0.1.7
jmp>=0.0.3
fsspec[http]<2024.7,>=2024.2
tensorstore>=0.1.62
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

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