Flax

by Google

JAX neural-net library used by whisper-jax.

TL;DR

JAX neural-net library used by whisper-jax.

Best for training Whisper-jax forks + custom JAX speech models. Pricing: free.

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

What it is

Google Brain's JAX neural-net library. Apache-2.0.

Best for: Training Whisper-jax forks + custom JAX speech models.
Watch out for: Smaller community than PyTorch.

Install / use

pip install flax

Features

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

Flax vs Whipscribe

FeatureFlaxWhipscribe
CategoryOpen sourceTranscription APIs
Pricingfreefree beta
Speaker diarizationNoYes
Word timestampsYesYes
StreamingNoNo
Languages9999
PlatformsLinux, macOS, TPUWeb, API, MCP

Alternatives to Flax

OpenAI Whisper
OpenAI

The reference open-source multilingual ASR model from OpenAI.

OSS · MIT ★ 98.1k
whisper.cpp
Georgi Gerganov

C/C++ port of Whisper — runs on anything, from a Raspberry Pi to Apple Silicon.

OSS · MIT ★ 48.8k
faster-whisper
SYSTRAN

4× faster than reference Whisper using CTranslate2 — production sweet spot.

OSS · MIT ★ 22.3k

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.