Google's custom AI chips, the tensor processing units (TPUs), are now serving AI workloads in production at Meta through a new PyTorch-native version of the SGLang inference engine, according to an announcement on Google's developer blog. A matching version of the vLLM inference engine was also announced as open source. Previously, both engines used Google's JAX AI framework to run on TPUs.
Meta's PyTorch-TPU lead Zongwei Zhou said SGLang-TorchTPU is "running in production at Meta today" after "a close three-way collaboration between Meta, Google, and the SGLang team." The team is based at RadixArk, a startup founded by SGLang's creators. Zhou described the engine as giving Meta "a remarkably mature, PyTorch-native serving stack."
Jana van Greunen, director of PyTorch engineering at Meta, said the company "cannot be constrained by a single hardware architecture" and that TorchTPU "enables us to deploy our PyTorch workloads to Google TPUs." Google said it co-developed TorchTPU with Meta's PyTorch team.
Google called PyTorch/XLA, the older bridge from PyTorch to TPUs, a "predecessor" to the new TorchTPU. SemiAnalysis reported last year that Meta research teams used TPUs through that bridge from 2020 to 2023 before canceling those contracts. While it is unclear where Meta's production TPUs are located, reports last year indicated talks about Google Cloud TPU rentals in 2026 with purchases in 2027, and Meta later reportedly signed a multi-year rental agreement.
The announcement came one day after Amazon said it would open-source TorchNeuron, its own native PyTorch backend for its Trainium2 and Trainium3 chips, at the PyTorch Conference.
On performance, a tuned vLLM setup serving Moonshot AI's open-weight, 2.8-trillion-parameter Kimi K3 model reportedly tops 700 tokens per second for a single user on the seventh-generation Ironwood TPU, according to Inferact, the startup founded by vLLM's creators that helped Google build vLLM-TorchTPU. That test ran 16 Ironwood chips with speculative decoding, compared to vLLM's published setup on 16 Nvidia GB200s, which reached 452 tokens per second. Kernels for Kimi K3 and Qwen 3.8 27B deliver roughly 1.4 to 2 times the GB200 baseline's throughput at small batch sizes without speculative decoding.
SGLang-TorchTPU will keep adding models, starting with DeepSeek V4.1 Flash and Reflection Beam. RadixArk said it is working toward Day 0 TPU support for new open models by the first quarter of 2027. Google, Meta and RadixArk plan to share more at SGLang Summit 2026 in San Francisco on Nov. 12-13.
