whisper-jax
JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.
GraphCanon updated Sep 20, 2026 · GitHub synced Sep 20, 2026
51views 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 (899d since push)
- As of Sep 20, 2026
- Provenance
- Not a fork · Personal account
- As of Sep 20, 2026
- Security (OSV)
- No lockfile
- As of Jul 11, 2026
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 · Sep 20, 2026
Categories
Compatibility
Sourced claims from the README excerpt - not unsourced marketing copy.
Source: README excerpt (regex_v1, Sep 20, 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 installati...
For agents
This page has a .md twin and JSON over the API.