Skip to content

Optimize Wan VAE: Add Temporal Chunking, On-TPU Postprocessing, Static JIT Support, and Configurable Dtypes#411

Merged
copybara-service[bot] merged 1 commit into
mainfrom
wan_vae_opt
May 26, 2026
Merged

Optimize Wan VAE: Add Temporal Chunking, On-TPU Postprocessing, Static JIT Support, and Configurable Dtypes#411
copybara-service[bot] merged 1 commit into
mainfrom
wan_vae_opt

Optimize and scale WAN VAE spatial sharding

1710495
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded May 26, 2026 in 13s

✅ 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:

1710495 Author: @Perseus14 <rish********nlc​@gmail.com>

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