Whisper JAX
Whisper JAX code for OpenAI's Whisper Model, largely built on the 🤗 Hugging Face Transformers Whisper implementation. Compared to OpenAI's PyTorch code, Whisper JAX runs over 70x faster, making it the fastest Whisper implementation available.

The JAX code is compatible on CPU, GPU and TPU, and can be run standalone on a Cloud TPU.

For a quick-start guide to running Whisper JAX on a Cloud TPU, refer to the following Kaggle notebook, where we transcribe 30 mins of audio in approx 30 sec:

Sachit Gandhi's Kaggle Notebook - [ Ссылка ]

Whisper JAX Web UI - [ Ссылка ]

❤️ If you want to support the channel ❤️
Support here:
Patreon - [ Ссылка ]
Ko-Fi - [ Ссылка ]

свернуть