Skip to content

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

0f430e3
Select commit
Loading
Failed to load commit list.
Open

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

Fix XLA stall: avoid per-batch .item() for loss accumulation
0f430e3
Select commit
Loading
Failed to load commit list.
Meta CLA / Meta CLA Check succeeded Mar 30, 2026 in 0s

Contributor License Agreement is valid!