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

optimum-tpu

archived
huggingface/optimum-tpu

Google TPU optimizations for transformers models

GraphCanon updated Oct 3, 2026 · GitHub synced Oct 3, 2026

37views this month

136 stars30 forksLast push Jan 23, 2026 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 (252d since push)
As of Oct 3, 2026
Provenance
Not a fork · Organization account
As of Oct 3, 2026
Security (OSV)
1 critical, 3 medium, 17 low (1 critical, 3 medium, 17 low)
As of Jul 11, 2026

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·Jul 11, 2026
Employees
160·Wikidata (P1128 employees)·Jul 11, 2026
Funding
$235,000,000 (2023-08)·GraphCanon curated seed (public press)·Jul 11, 2026
Commercial model
OSS + managed cloud·GraphCanon curated seed·Jul 11, 2026

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] · Oct 3, 2026

Languages
python

Source: github.language+pyproject.toml · Oct 3, 2026

Categories

Graph entities

Compatibility

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

Python runtimePython

Source: README excerpt (regex_v1, Oct 3, 2026)

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

Tags

README

Installation comes with an handy PyPi released package compatible with your classical python dependency management tool.

For agents

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

Was this helpful?

Anonymous feedback helps us improve pages and translations.