JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.
Repository Index
88,073 visible repositories in the index
1–3 of 3 · Most stars
Runs 405B LLMs on 8GB VRAM
2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.