JetStream

Throughput and memory optimized engine for LLM inference on XLA devices.

GrowingOpen SourceLow lock-in

Pricing

Free tier

Flat rate

Adoption

Cooling

License

Open Source

Data freshness

Aging · Jun 8, 2026

Overview

What is JetStream?

JetStream is a throughput and memory-optimized engine designed for Large Language Model (LLM) inference, specifically targeting XLA devices like TPUs. It aims to enhance performance by optimizing resource usage, making it ideal for high-throughput scenarios.

Key differentiator

JetStream stands out as an open-source, high-throughput engine optimized for TPU devices, offering efficient memory usage and performance tuning.

Capability profile

Capability Radar

Ease of StartEcosystemValueMaturityFlexibilityScale Ready

Honest assessment

Strengths & Weaknesses

↑ Strengths

Optimized for high throughput and memory efficiency on XLA devices.medium

Initial support for TPUs with future plans for GPU support.medium

Open-source under Apache-2.0 license.medium

↓ Weaknesses

Limited device support currently only for TPUshigh

Documentation and examples are heavily focused on TPU usage, with no official GPU support yet.

Complex setup process for non-expert usersmedium

Setup requires detailed configuration of XLA devices which can be challenging without deep knowledge of hardware and software stack.

Poor documentation for advanced use caseshigh

The official documentation lacks comprehensive guides on optimizing performance beyond basic setups, leaving users to infer details from source code or community forums.

Performance degradation with non-XLA compatible hardwaremedium

JetStream's optimizations are specifically tailored for XLA devices and may not yield the same performance benefits on other types of hardware without significant customization.

Fit analysis

Who is it for?

✓ Best for

Teams working on high-throughput applications that require efficient use of TPUs.

Projects focused on optimizing memory and throughput in LLM inference tasks.

✕ Not a fit for

Applications requiring real-time streaming capabilities (batch-only architecture).

Scenarios where GPU support is a must-have feature at present.

Cost structure

Pricing

Free Tier

Available

Open source — free to use

Starts at

$0

Model

Flat rate

Enterprise

None

Performance benchmarks

How Fast Is It?

Ecosystem

Relationships

Next step

Get Started with JetStream

Step-by-step setup guide with code examples and common gotchas.

View Setup Guide →