librosa
pytest
scipy
numpy
tqdm
jax
jaxlib
flax