chex>=0.1.86
einops
flax>=0.8.2
grain==0.2.*
librosa>=0.10.1
numpy
soundfile
typing_extensions

[dev]
pytest
pytest-cov
