Flax is a neural network library for JAX that is designed for flexibility.
Repository Index
47,133 repositories indexed
1–2 of 2 · Most stars
JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.
47,133 repositories indexed
1–2 of 2 · Most stars
Flax is a neural network library for JAX that is designed for flexibility.
JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.