Search results

RNN / GRU / LSTM implementation for torch_xla

Status: Open.

#8655 In pytorch/xla;

· qihqi opened on Jan 30, 2025

installing torch_xla 2.6.0 does not work

Status: Open.

#8652 In pytorch/xla;

· ttdd11 opened on Jan 30, 2025

how to use device_map in diffusers.stablediffusionpipeline.from_pretrained

Status: Open.

#8646 In pytorch/xla;

· chaowenguo opened on Jan 29, 2025

Make Mixtral pallas kernels Dynamo/AOTAutograd traceable

Status: Open.

#8642 In pytorch/xla;

· tengyifei opened on Jan 28, 2025

split on second dimension of 2D array not working

Status: Open.

#8640 In pytorch/xla;

· jeffhataws opened on Jan 28, 2025

[torchax][RFC] Anchor on the device API everywhere

RFC

torchxla2

Status: Open.

#8638 In pytorch/xla;

· tengyifei opened on Jan 28, 2025

[torchax] RNG handling in a jitted graph is unsound

Status: Open.

#8636 In pytorch/xla;

· tengyifei opened on Jan 28, 2025

[torchax] jit compile the model constructor

Status: Open.

#8635 In pytorch/xla;

· tengyifei opened on Jan 28, 2025

Make flash_attention Dynamo/AOTAutograd traceable

Status: Open.

#8633 In pytorch/xla;

· tengyifei opened on Jan 27, 2025

[scan] Avoid re-tracing the combine function on every call

Status: Open.

#8632 In pytorch/xla;

· tengyifei opened on Jan 27, 2025

Add type annotation to PyTorch/XLA code and tests & expand tests for various types as needed

TECHNICAL_DEBTTechnical Debt Is EvilTechnical Debt Is Evil

usabilityBugs/features related to improving the usability of PyTorch/XLABugs/features related to improving the usability of PyTorch/XLA

Status: Open.

#8627 In pytorch/xla;

· miladm opened on Jan 26, 2025

XLA 2.7 releases don't work properly with the upstream torch 2.7

Status: Open.

#8626 In pytorch/xla;

· hosseinsarshar opened on Jan 26, 2025