Search results

SPMD: spurious retrace every step due to sharding spec mismatch for unsharded tensors

Status: Open.

#9755 In pytorch/xla;

· ajakovljevicTT opened on Mar 2, 2026

Optimization

Status: Open.

#9754 In pytorch/xla;

· signore662-beep opened on Mar 1, 2026

Is torch_xla going to be deprecated soon?

Status: Open.

#9752 In pytorch/xla;

· grib0ed0v opened on Feb 17, 2026

Why does ReplicateShardedData perform an uncached compile+device execute on each sharded output to return to host

Status: Open.

#9751 In pytorch/xla;

· jameszianxuTT opened on Feb 4, 2026

Why might my program hang indefinitely when I use torch_xla.sync?

Status: Open.

#9750 In pytorch/xla;

· oluwatimilehin opened on Feb 2, 2026

[Bug] Calling torch_xla.device() or setting random seed before use_spmd() produces SIGSEGV for unmarked tensors

Status: Open.

#9735 In pytorch/xla;

· Dogacel opened on Jan 20, 2026

reduce memory requirements for FSDP

Status: Open.

#9734 In pytorch/xla;

· samuele-bortolato-mag opened on Jan 16, 2026

Torchax registering jax as PyTorch accelerator

Status: Open.

#9730 In pytorch/xla;

· ajakovljevicTT opened on Dec 15, 2025

Add priority-aware sharding via mark_sharding to guide Shardy propagation

Status: Open.

#9729 In pytorch/xla;

· sshonTT opened on Dec 12, 2025

all_to_all operation generates invalid HLO

bugSomething isn't workingSomething isn't working

distributedSPMD and other distributed things.SPMD and other distributed things.

Status: Open.

#9727 In pytorch/xla;

· rajkthakur opened on Dec 5, 2025

How is GetOutputShardings supposed to work for PJRT Implementers?

question

runtime

stablehloStableHLO related workStableHLO related work

Status: Open.

#9726 In pytorch/xla;

· jameszianxuTT opened on Dec 2, 2025

Why isn't there a binding for clearing the XLAGraphExecutor::ComputationCache?

question

runtime

Status: Open.

#9712 In pytorch/xla;

· jameszianxuTT opened on Nov 14, 2025