TorchTPU
Named by 8 kept items in this workspace.
Search every workspace for TorchTPU. It matches the name as written, so it can find other things that share it.
The items that name it 8
Google's open-source native PyTorch path for training and serving on TPUs.
Open-sourced by Google to run PyTorch apps on its TPUs. Google says moving an app takes a simple configuration change with no rewrite. It comes with an agent suite that helps migrate apps and fix errors.
Native PyTorch backend that adds TPU as a device. It includes Pallas kernels tuned for TPUs, distributed training support and torch.compile integration. Installed with pip install torch-tpu.
Used to serve TPUv7 AgentX runs with vLLM.
Used to serve vLLM on TPUv7
Used with vLLM at FP8 on TPUv7 in SemiAnalysis benchmark
- [AINews] AEF-1 standard emerges for Third Party Evaluators, as Xai, OpenAI, and Anthropic all cosign
Native PyTorch path for running on TPU as part of the vLLM/TPU integration
Native PyTorch backend that exposes TPUs as device="tpu", lowering through StableHLO and XLA. Intended as the backend for vLLM and SGLang on TPU. In private beta in early September, with open-sourcing expected around mid-October at PyTorch Conference.
All technologies in this workspace.