chex>=0.1.86
einops
flax>=0.8.2
grain==0.1.0
librosa>=0.10.1
numpy
soundfile

[dev]
pytest
pytest-cov
