optimum-tpu

Optimum-TPU is a Hugging Face project providing Google TPU optimizations for transformers models, aiming to ease training and inference on TPUs. It supports text generation models including Gemma, Llama2, Llama3 and Mistral, and includes a text-generation-inference backend. The repository was archived in January 2026 and is read-only, with the project in maintenance mode.

GitHub repo Directory import
0 shouts Added Sep 23, 2026 github.com
github.com/huggingface/optimum-tpu
optimum-tpu website preview

Features

Links

Tags