GraphCanon
Home/Speech & Audio/whisper-jax
whisper-jax logo

whisper-jax

Enrichment pending
sanchit-gandhi/whisper-jax

JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.

GitHub

GraphCanon updated today · GitHub synced today

4.7k
Stars
413
Forks
140
Open issues
42
Watchers
2y
Last push
Jupyter Notebook Apache-2.0Created Mar 2, 2023

Trust & integrity

Full report
Maintenance
Dormant (829d since push)
As of today · Source: github_public_v1
Provenance
Not a fork · Personal account
As of today · Source: github_public_v1
Security (OSV)
No lockfile
As of today · Source: none

Public GitHub metadata and optional OSV dependency scans. Signals, not a guarantee. Trust methodology.

Overview

JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.

Capability facts

Languages
jupyter notebook, python

Source: github.language+pyproject.toml · Jul 11, 2026

Categories

Speech & Audio

Compatibility

Sourced claims from the README excerpt - not unsourced marketing copy.

Python runtimePython

Source: README excerpt (regex_v1, Jul 11, 2026)

Whisper JAX was tested using Python 3.9 and JAX version 0.4.5. Installation assumes that you already have the latest
Source link

Tags

deep-learningspeech-to-textjupyter notebookspeech-recognitionwhisperjax

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