Sanchit Gandhi's repositories
whisper-jax
JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.
seq2seq-speech
Repository for fine-tuning Transformers 🤗 based seq2seq speech models in JAX/Flax.
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.