S5 / Essay_classifier /requirements_gpu.txt
dbal0503's picture
Upload 693 files
2ce7b1a
raw
history blame
170 Bytes
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