JAX (Google)

by Google

Underlying framework for whisper-jax and TPU ASR research.

TL;DR

Underlying framework for whisper-jax and TPU ASR research.

Best for training / serving custom ASR research stacks on TPU. Pricing: free.

Category
Open source
License
Stars
Last push
Pricing
free
Platforms
Linux, macOS, TPU

What it is

Google's autograd + XLA composable transformations library. Apache-2.0.

Best for: Training / serving custom ASR research stacks on TPU.
Watch out for: Steeper than PyTorch for newcomers.

Install / use

pip install jax

Features

Speaker diarizationNo
Word-level timestampsYes
Streaming / real-timeNo
Languages supported99
HIPAA eligibleNo

JAX (Google) vs Whipscribe

FeatureJAX (Google)Whipscribe
CategoryOpen sourceTranscription APIs
Pricingfreefree beta
Speaker diarizationNoYes
Word timestampsYesYes
StreamingNoNo
Languages9999
PlatformsLinux, macOS, TPUWeb, API, MCP

Alternatives to JAX (Google)

Whipscribe is a managed faster-whisper + whisperX service. If you want transcripts without running infrastructure, paste a URL or drop a file in the form below — you'll have a transcript in seconds.