Optimized implementation of the Whisper model.   Link: Whisper JAX