flax
torch
torchtext
tensorflow-datasets==4.5.2
pydub==0.25.1
datasets
tqdm
--find-links https://storage.googleapis.com/jax-releases/jax_releases.html
jax[cuda]>=version