RepoJournal
PyTorch

@pytorch

PyTorch and the broader machine-learning ecosystem

Keep up with PyTorch in about 3 minutes a day: what actually shipped — the commits, pull requests, releases, and security advisories that matter.

One email a day. Unsubscribe in one click.

Pick a date

Topics: Python AI / ML Full archive →

The Wire · Showcase

TORCHTITAN SHIPS FLEXATTENTION INDUCTOR BOOST, GRAPH TRAINER UNLOCKS CPU OFFLOADING

By RepoJournal · Filed · About PyTorch

FlexAttention now compiles through Inductor when using aot_eager backend, cutting Step 1 training loss mismatch significantly while the graph trainer gains view replay for CPU-offloaded activations.

The regional_inductor context manager [1] wraps FlexAttention ops to trigger Inductor compilation instead of falling back to eager, validated on RL workloads where Step 1 loss variance dropped measurably. This pairs with a major fix in graph_trainer [2] that replays view operations (transpose, reshape, permute) during backward, finally enabling CPU activation offloading for tensors whose consumers reach them through view chains. Qwen3.5 evolution [3] shipped with hybrid attention architecture (75% GatedDeltaNet linear + 25% full attention) and head-sharded TP on GatedDeltaNet projections, marking a significant architecture jump from Qwen3-VL. The RL infrastructure expanded with a GeneratorRouter [4] supporting round-robin and least-loaded routing across multiple generators for large-scale training, plus weight sync modes for hot-swap deployment. On the PyTorch core side, the build system fixed a critical bug in build_with_debinfo.py [5] that broke targeted debug builds with CONFIGURE_DEPENDS globbing, while Dynamo now serializes higher-order-op subgraphs correctly [6] so fx_graph_runnable repros work for cond/while_loop branches. Inductor's assertion removal [7] [8] continues hardening error handling across fx_passes.

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] [RL] Enable regional_inductor in FlexAttention ↗ pytorch/torchtitan
  2. [2] [graph_trainer] Add view replay for CPU activation offloading ↗ pytorch/torchtitan
  3. [3] [qwen3_5] evolve qwen3_vl to qwen3_5 ↗ pytorch/torchtitan
  4. [4] Add a router for multiple generators ↗ pytorch/torchtitan
  5. [5] Fix build_with_debinfo.py broken by CONFIGURE_DEPENDS globbing (#186780) pytorch/pytorch
  6. [6] [dynamo] Serialize higher-order-op subgraphs in NNModuleToString.convert (#186804) pytorch/pytorch
  7. [7] remove plain assertions in remaining torch/_inductor top-level files (#186392) pytorch/pytorch
  8. [8] remove plain assertions in torch/_inductor/fx_passes (#186391) pytorch/pytorch

Quick answers

What shipped in PyTorch on June 11, 2026?
FlexAttention now compiles through Inductor when using aot_eager backend, cutting Step 1 training loss mismatch significantly while the graph trainer gains view replay for CPU-offloaded activations. In total, 99 commits and 40 pull requests landed.
Who contributed to PyTorch on June 11, 2026?
4 developers shipped this update, including aditvenk, mlazos, shuhuayu, and pzhan9.
What were the notable PyTorch updates?
[RL] Enable regional_inductor in FlexAttention, [graph_trainer] Add view replay for CPU activation offloading, and [qwen3_5] evolve qwen3_vl to qwen3_5.

More from @pytorch

Daily updates, in your inbox

Follow PyTorch

Keep up with PyTorch in about 3 minutes a day: what actually shipped — the commits, pull requests, releases, and security advisories that matter.

or

One email a day. Unsubscribe in one click. Read a past issue →

Elsewhere on the wire

Want every project, not just this one?

We use privacy-friendly analytics (Google Analytics, IP-anonymized) to see which pages help readers. No ads, and we never sell your data. See our Privacy Policy.