Skip to content

perf(wan2.2): Optimize attention kernels, rematerialization, and Ulysses parallelism for WAN 2.2 Training on TPU 7x - #487

Merged
copybara-service[bot] merged 1 commit into
AI-Hypercomputer:mainfrom
Toshi-31:wan-2.2-perf-optimizations
Oct 12, 2026
Merged

perf(wan2.2): Optimize attention kernels, rematerialization, and Ulysses parallelism for WAN 2.2 Training on TPU 7x#487
copybara-service[bot] merged 1 commit into
AI-Hypercomputer:mainfrom
Toshi-31:wan-2.2-perf-optimizations

perf(wan2.2): Optimize attention kernels, rematerialization, and Ulys…

069e7fb
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Oct 1, 2026 in 14s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

✅ 069e7fb PR Opener: @Toshi-31
✅ 069e7fb Author: <tos******dia​@google.com>

(Only the first commit for a unique contributor is listed.)

View more details on Google CLA