No project description provided
Project description
| Documentation | Blog | User Forum | Developer Slack (#sig-tpu) |
Latest News 🔥
-
Pytorch Conference Learn how Spotify uses vLLM with both GPUs and TPUs to drive down costs and improve user experience.
-
Check back soon for a recording of our session at Ray Summit, November 3-5 in San Francisco!
-
Check back soon for a recording of our session at JAX DevLab on November 18th in Sunnyvale!
-
[2025/10] vLLM TPU: A New Unified Backend Supporting PyTorch and JAX on TPU
Previous News 🔥
About
vLLM TPU is now powered by tpu-inference, an expressive and powerful new hardware plugin unifying JAX and PyTorch under a single lowering path within the vLLM project. The new backend now provides a framework for developers to:
- Push the limits of TPU hardware performance in open source.
- Provide more flexibility to JAX and PyTorch users by running PyTorch model definitions performantly on TPU without any additional code changes, while also extending native support to JAX.
- Retain vLLM standardization: keep the same user experience, telemetry, and interface.
Recommended models and features
Although vLLM TPU’s new unified backend makes out-of-the-box high performance serving possible with any model supported in vLLM, the reality is that we're still in the process of implementing a few core components.
For this reason, we’ve provided a Recommended Models and Features page detailing the models and features that are validated through unit, integration, and performance testing.
Get started
Get started with vLLM on TPUs by following the quickstart guide.
Visit our documentation to learn more.
Compatible TPU Generations
- Recommended: v5e, v6e
- Experimental: v3, v4, v5p
Check out a few v6e recipes here!
Contribute
We're always looking for ways to partner with the community to accelerate vLLM TPU development. If you're interested in contributing to this effort, check out the Contributing guide and Issues to start. We recommend filtering Issues on the good first issue tag if it's your first time contributing.
Contact us
- For technical questions and feature requests, open a GitHub Issue
- For feature requests, please open one on Github here
- For discussing with fellow users, use the TPU support topic in the vLLM Forum
- For coordinating contributions and development, use the Developer Slack
- For collaborations and partnerships, contact us at vllm-tpu@google.com
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file tpu_inference-0.13.3.tar.gz.
File metadata
- Download URL: tpu_inference-0.13.3.tar.gz
- Upload date:
- Size: 510.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
0621111aad38bb4a4fe7130a11011ebb0dbb873f87357f4bcab4aae1c9d14edb
|
|
| MD5 |
299970ff48f2bcc9437efdc21d22ae74
|
|
| BLAKE2b-256 |
beb675fbd0f8b26c6237c52604ef9c37ba4bc3170402442302410fae26a05c39
|
File details
Details for the file tpu_inference-0.13.3-py3-none-any.whl.
File metadata
- Download URL: tpu_inference-0.13.3-py3-none-any.whl
- Upload date:
- Size: 690.3 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
11b7dfa04830c96dcbd41c24988445aa49302c33a9859707ea67852bfccc0160
|
|
| MD5 |
9d633d43affee36ba2d7072a875e4884
|
|
| BLAKE2b-256 |
475bc6093f3503a115d58cab5c473d52230213d73d9bab168ce2739257224738
|
