Skip to content

vae example: provide an additional --xla flag to be able to run this test on TPU if available#1416

Open
fedebongio wants to merge 2 commits into
pytorch:mainfrom
fedebongio:vae-xla-support
Open

vae example: provide an additional --xla flag to be able to run this test on TPU if available#1416
fedebongio wants to merge 2 commits into
pytorch:mainfrom
fedebongio:vae-xla-support

Fix XLA stall: avoid per-batch .item() for loss accumulation

0f430e3
Select commit
Loading
Failed to load commit list.
Netlify / Header rules completed Mar 30, 2026 in 3s

Deploy canceled

Please check the logs.

View more details on Netlify