{"data":{"slug":"sanchit-gandhi-whisper-jax","name":"whisper-jax","tagline":"JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.","github_url":"https://github.com/sanchit-gandhi/whisper-jax","owner":"sanchit-gandhi","repo":"whisper-jax","owner_avatar_url":"https://avatars.githubusercontent.com/u/93869735?v=4","primary_language":"Jupyter Notebook","stars":4680,"forks":410,"topics":["deep-learning","jax","speech-recognition","speech-to-text","whisper"],"archived":false,"github_pushed_at":"2024-04-03T12:12:52+00:00","maintenance_label":"Dormant","stars_delta_30d":-4,"url":"https://www.graphcanon.com/tools/sanchit-gandhi-whisper-jax","markdown_url":"https://www.graphcanon.com/tools/sanchit-gandhi-whisper-jax.md","api_url":"https://www.graphcanon.com/api/graphcanon/tools/sanchit-gandhi-whisper-jax","graph_url":"https://www.graphcanon.com/api/graphcanon/graph?tool=sanchit-gandhi-whisper-jax","description":"JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.","homepage_url":null,"license":"Apache-2.0","open_issues":138,"watchers":45,"ai_summary":"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.","readme_excerpt":"## Installation\n\nWhisper JAX was tested using Python 3.9 and JAX version 0.4.5. Installation assumes that you already have the latest \nversion of the JAX package installed on your device. You can do so using the official JAX installation guide: https://github.com/google/jax#installation\n\nOnce the appropriate version of JAX has been installed, Whisper JAX can be installed through pip:\n```\npip install git+https://github.com/sanchit-gandhi/whisper-jax.git\n```\n\nTo update the Whisper JAX package to the latest version, simply run:\n```\npip install --upgrade --no-deps --force-reinstall git+https://github.com/sanchit-gandhi/whisper-jax.git\n```","github_created_at":"2023-03-02T12:30:52+00:00","created_at":"2026-07-11T12:15:24.930803+00:00","updated_at":"2026-09-20T07:17:10.171765+00:00","categories":[{"slug":"inference-serving","name":"Inference & Serving","url":"https://www.graphcanon.com/categories/inference-serving","markdown_url":"https://www.graphcanon.com/categories/inference-serving.md","api_url":"https://www.graphcanon.com/api/graphcanon/categories/inference-serving"},{"slug":"speech-audio","name":"Speech & Audio","url":"https://www.graphcanon.com/categories/speech-audio","markdown_url":"https://www.graphcanon.com/categories/speech-audio.md","api_url":"https://www.graphcanon.com/api/graphcanon/categories/speech-audio"}],"tags":[{"slug":"deep-learning","name":"deep-learning"},{"slug":"jax","name":"jax"},{"slug":"speech-recognition","name":"speech recognition"},{"slug":"speech-to-text","name":"speech-to-text"},{"slug":"whisper","name":"whisper"}],"trust":{"provenance":{"is_fork":false,"github_id":608634021,"owner_type":"User","methodology":"github_public_v1","parent_repo":null,"near_duplicate_slugs":[]},"computed_at":"2026-09-20T07:17:08.566Z","maintenance":{"label":"Dormant","score":18,"methodology":"github_public_v1","releases_90d":0,"days_since_push":899,"last_release_at":null,"stars_delta_30d":-4,"open_issues_delta_30d":-2},"security_summary":{"status":"no_lockfile","scanner":null,"low_count":0,"high_count":0,"last_scan_at":"2026-07-11T12:15:34.365Z","medium_count":0,"scan_profile":"none","critical_count":0}},"capability_facts":{"scan":{"source":"repo_scan","observed_at":"2026-09-20T07:17:09.634Z"},"languages":{"value":["jupyter notebook","python"],"source":"github.language+pyproject.toml","observed_at":"2026-09-20T07:17:09.634Z"},"license_spdx":{"value":"Apache-2.0","source":"github.license","observed_at":"2026-09-20T07:17:09.634Z"}},"decision_facts":{"hosting":null,"pricing":null,"requirements":{"notes":["- Python 3.9 and JAX version 0.4.5 are mandatory requirements to ensure compatibility with whisper-jax."]},"constraints":null,"when_to_use":["- 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."],"when_not_to_use":["- 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."],"source":"enrich:decision_facts","observed_at":"2026-07-17T01:10:14.432Z"},"constraint_facets":null,"decision_summary":[{"label":"Requirements","value":"- Python 3.9 and JAX version 0.4.5 are mandatory requirements to ensure compatibility with whisper-jax."},{"label":"Adopt for","value":"whisper-jax is a JAX-based implementation of OpenAI's Whisper model that delivers up to 70x speed improvement when executed on Google TPUs."},{"label":"License detail","value":"Licensed under Apache-2.0, offering permissive terms for use in both commercial and non-commercial contexts."}]}}