Home/Model Training/optimum-tpu
optimum-tpu logo

optimum-tpu

archived
huggingface/optimum-tpu

Google TPU optimizations for transformers models

GraphCanon updated 2w · GitHub synced 2w

135 stars30 forksLast push 7mo Python Apache-2.0

Decision brief

optimum-tpu is tailored for Python developers working with transformers models aiming to leverage the power of Google TPUs.

Good fit when

  • Use optimum-tpu when you require high performance execution of transformers models on Google TPUs, as it offers specific optimizations for that hardware.
  • Ideal for teams already invested in the Google Cloud ecosystem where TPU access is seamless and integrated.

Avoid when

  • Avoid using optimum-tpu if your infrastructure does not include or will not support Google TPUs, since its optimizations are not beneficial on other hardware.
  • Skip this tool if you are working in environments with strict licensing requirements as it requires adherence to the Apache-2.0 license.
Hosting:
self hosted
Pricing:
freemium

Observed Jul 16, 2026 · Source: enrich:decision_facts

Verify the decision

Maintenance and security

Full trust report
Maintenance
Archived (193d since push)
As of 2w
Provenance
Not a fork · Organization account
As of 2w
Security (OSV)
1 critical, 3 medium, 17 low (1 critical, 3 medium, 17 low)
As of 1mo

Public GitHub metadata and optional OSV scans. Signals, not a guarantee. Trust methodology.

Backing

Company context for Hugging Face. Display-only - separate from trust and ranking.

Company
Hugging Face·GitHub org profile·1mo
Employees
160·Wikidata (P1128 employees)·1mo
Funding
$235,000,000 (2023-08)·GraphCanon curated seed (public press)·1mo
Commercial model
OSS + managed cloud·GraphCanon curated seed·1mo

Install

pip install optimum-tpu
PyPI

Similar tools

Same-category neighbours. No typed graph edges are catalogued for this tool yet.

Evidence and technical details

Sourced facts, taxonomy, compatibility claims, README excerpt, and machine-readable endpoints.

Overview

A Python library that optimizes transformers models for execution on Google TPUs.

Capability facts

CLI
CLI entrypoint

Source: pyproject.toml:[project.scripts] · Aug 4, 2026

Languages
python

Source: github.language+pyproject.toml · Aug 4, 2026

Categories

Graph entities

Compatibility

Sourced claims from the README excerpt - not unsourced marketing copy.

Python runtimePython

Source: README excerpt (regex_v1, Aug 4, 2026)

m-tpu` comes with an handy PyPi released package compatible with your classical python dependency management tool.
Source link

Tags

README

Installation

optimum-tpu comes with an handy PyPi released package compatible with your classical python dependency management tool.

pip install optimum-tpu -f https://storage.googleapis.com/libtpu-releases/index.html

export PJRT_DEVICE=TPU

For agents

This page has a .md twin and JSON over the API.

Was this helpful?

Anonymous feedback helps us improve pages and translations.