sanchit-gandhi/whisper-jax

This is stars and forks stats for /sanchit-gandhi/whisper-jax repository. As of 29 Apr, 2024 this repository has 3450 stars and 273 forks.

Whisper JAX This repository contains optimised 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 (see Pipeline Usage) or as an inference endpoint (see Creating an Endpoint). For a quick-start guide to running Whisper JAX on a Cloud TPU, refer...