Search results

Unable to build torch/xla

Status: Open.

#9454 In pytorch/xla;

· mikegre-google opened on Jul 8, 2025

torch_xla.tpu.version() gets stuck occasionally

bugSomething isn't workingSomething isn't working

xla:tpuTPU specific issues and PRsTPU specific issues and PRs

Status: Open.

#9449 In pytorch/xla;

· yaochengji opened on Jul 7, 2025

[RFC] Controller for SPMD+MPMD

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

RFC

Status: Open.

#9447 In pytorch/xla;

· pgmoka opened on Jul 7, 2025

2.8 backport PR request list

backport2.8Tag for PRs that need to be backported in r2.8Tag for PRs that need to be backported in r2.8

Status: Open.

#9433 In pytorch/xla;

· pgmoka opened on Jul 1, 2025

[RFC] Enable DTensor SPMD APIs with XLA SPMD Backend

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

enhancementNew feature or requestNew feature or request

RFC

Status: Open.

#9418 In pytorch/xla;

· fhaolinaws opened on Jun 27, 2025

[TORCHAX] Sharded jax tensors support for FlaxNNModule

enhancementNew feature or requestNew feature or request

torchxla2

Status: Open.

#9413 In pytorch/xla;

· vlad-karp opened on Jun 26, 2025

[TORCHAX] Can't allocate random tensors with device='jax'

bugSomething isn't workingSomething isn't working

torchxla2

Status: Open.

#9411 In pytorch/xla;

· vlad-karp opened on Jun 26, 2025

Changes needed for <code>device_assignment</code> in <code>PjRtComputationClient::Compile</code> to support submeshing/localized spmd

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

enhancementNew feature or requestNew feature or request

Status: Open.

#9391 In pytorch/xla;

· kvshbg-aws opened on Jun 18, 2025

Create and Expose the <code>torch_xla::OpSharding</code> wrapper class instead of <code>xla::OpSharding</code> class

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

enhancementNew feature or requestNew feature or request

Status: Open.

#9390 In pytorch/xla;

· kvshbg-aws opened on Jun 18, 2025

Normalize <code>tile_assignment</code> after constructing the <code>xla::OpSharding</code> object

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

enhancementNew feature or requestNew feature or request

Status: Open.

#9389 In pytorch/xla;

· kvshbg-aws opened on Jun 18, 2025

Update pins for week of June 16, 2025

dependenciesPull requests that update a dependency filePull requests that update a dependency file

enhancementNew feature or requestNew feature or request

Status: Open.

#9387 In pytorch/xla;

· yaoshiang opened on Jun 18, 2025

[v2.8] Neuron inference trace analyzer/bucketing unit tests hanging at GetParameterIdTensorMapping/TransferFromDevice

bugSomething isn't workingSomething isn't working

xla:neuron

Status: Open.

#9378 In pytorch/xla;

· jeffhataws opened on Jun 17, 2025