sanchit-gandhi/whisper-jax

Health v2 · maintenance only

Flags: no_releases

Adoption not part of the score

4682 stars · 411 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-29, confidence not recorded

An optimized JAX implementation of OpenAI's Whisper speech recognition model, built on Hugging Face Transformers, offering up to 70x faster transcription than the PyTorch version. It runs on CPU, GPU, and TPU with data parallelism via pmap and can be used as a pipeline or inference endpoint.

Use cases

• transcribe long audio files fast
• batch transcribe hours of audio on a TPU
• run whisper large v2 with jax
• deploy whisper as an inference endpoint
• speech to text for podcasts or meetings
• benchmark whisper implementations

When to choose

• you need maximum Whisper throughput on GPU/TPU
• you're transcribing large volumes of audio in batch
• you already use the JAX/Flax ecosystem

When to avoid

• you need a simple CPU-only transcription tool
• you require PyTorch ecosystem integration or fine-tuning tooling
• you need actively maintained features - the repo is in maintenance mode

Facets

library · maturity maintenance

speech-recognition machine-learning llm-inference speech-processing deep-learning machine-learning python cloud jax whisper tpu transcription asr hugging-face gpu

Member repositories

For agents

markdown · JSON · MCP: product_card(name="sanchit-gandhi/whisper-jax")