haliax>=1.4.dev343
equinox!=0.12.0,>=0.11.7
jaxtyping>=0.2.34
tokenizers>=0.15.2
transformers<5.0,>=4.49.0
optax>=0.1.9
wandb>=0.17.8
draccus>=0.11.5
pyarrow>=11.0.0
zstandard>=0.18.0
datasets<4.0,>=3.1.0
gcsfs<2026,>=2024.2
braceexpand>=0.1.7
jmp>=0.0.3
fsspec[http]<2026,>=2024.2
tensorstore>=0.1.73
pytimeparse>=1.1.8
humanfriendly==10.0
safetensors[numpy]<0.6.0,>=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
filelock~=3.13
async-lru~=2.0
tqdm-loggable>=0.2
deepdiff
lenses
jinja2

[test]
pytest
pytest-forked
pytest-asyncio
flake8
soundfile
librosa
tensorboardX>=2.6
