GraphCanon updated 2w · GitHub synced 2w
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 PyPISimilar 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.
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.