whisper-jax
JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.
GraphCanon updated 3w · GitHub synced 3w · 25 views this month
Decision brief
whisper-jax is a JAX-based implementation of OpenAI's Whisper model that delivers up to 70x speed improvement when executed on Google TPUs.
Good fit when
- - When aiming for high-performance inference, especially on Google TPU environments, whisper-jax offers significant acceleration.
- - For users familiar with JAX and leveraging Python 3.9 with a configured JAX environment version 0.4.5 or later.
Avoid when
- - If running in an environment that does not support JAX or lacks access to a Google TPU, as performance benefits may be minimal.
- - For users without existing setup for JAX and Python 3.9, the initial onboarding might involve additional effort.
- Requirements:
- - Python 3.9 and JAX version 0.4.5 are mandatory requirements to ensure compatibility with whisper-jax.
Observed Jul 17, 2026 · Source: enrich:decision_facts
Verify the decision
Maintenance and security
Full trust report- Maintenance
- Dormant (847d since push)
- As of 3w
- Provenance
- Not a fork · Personal account
- As of 3w
- Security (OSV)
- No lockfile
- As of 1mo
Public GitHub metadata and optional OSV scans. Signals, not a guarantee. Trust methodology.
Install
git clone https://github.com/sanchit-gandhi/whisper-jaxHow it fits your stack(1)
Typed graph edges - alternatives, integrations, successors, and dependencies. Ranked by relationship type, not raw GitHub stars.
Relationship graph
Optional deeper exploration of typed edges and category neighbours.
Similar tools
Same-category neighbours not already linked as typed edges.
Evidence and technical details
Sourced facts, taxonomy, compatibility claims, README excerpt, and machine-readable endpoints.
Overview
A JAX-based version of the Whisper speech recognition and transcription model optimized for high performance on Google TPUs, providing a significant speed improvement over other versions.
Capability facts
- Languages
- jupyter notebook, python
Source: github.language+pyproject.toml · Jul 30, 2026
Categories
Compatibility
Sourced claims from the README excerpt - not unsourced marketing copy.
Source: README excerpt (regex_v1, Jul 30, 2026)
Whisper JAX was tested using Python 3.9 and JAX version 0.4.5. Installation assumes that you already have the latestSource link
Tags
README
Installation
Whisper JAX was tested using Python 3.9 and JAX version 0.4.5. Installation assumes that you already have the latest version of the JAX package installed on your device. You can do so using the official JAX installation guide: https://github.com/google/jax#installation
Once the appropriate version of JAX has been installed, Whisper JAX can be installed through pip:
pip install git+https://github.com/sanchit-gandhi/whisper-jax.git
To update the Whisper JAX package to the latest version, simply run:
pip install --upgrade --no-deps --force-reinstall git+https://github.com/sanchit-gandhi/whisper-jax.git
For agents
This page has a .md twin and JSON over the API.