requirements.txt

transformers
pytube
nvidia-cublas-cu11
nvidia-cuda-nvcc-cu11
nvidia-cuda-runtime-cu11
nvidia-cudnn-cu11
nvidia-cufft-cu11
nvidia-cusolver-cu11
nvidia-cusparse-cu11
-f https://storage.googleapis.com/jax-releases/cuda11/jaxlib-0.4.7+cuda11.cudnn86-cp38-cp38-manylinux2014_x86_64.whl
git+https://github.com/sanchit-gandhi/whisper-jax.git