Pull Requests pytorch/xla

Support trace_me and xp.Trace in assume_pure

#9311 opened 2025-06-09 01:05 by tengyifei

Fix torchgen imports in codegen

#9310 opened 2025-06-07 20:35 by XuehaiPan

Stop tracking data and log files

#9308 opened 2025-06-06 22:22 by zzzwen

Fix race condition in runAtenTest

#9306 opened 2025-06-06 17:08 by benawilson

Implement prng_key as a mutable array

#9305 opened 2025-06-06 06:38 by tengyifei

Add an option to not use dlpack.

#9304 opened 2025-06-06 03:54 by qihqi

cleanup: remove defunct --distinct_host_configuration flag

#9300 opened 2025-06-05 20:58 by rickeylev

Add cache support for scan_layers

#9297 opened 2025-06-05 18:20 by iwknow

Add DLRM DCN v2 with JAX Sparsecore

#9287 opened 2025-06-03 20:01 by chandrasekhard2

WIP - do not review; Support python 3.12

#9274 opened 2025-06-01 16:24 by zhanyong-wan

doc: update pytorch-on-xla-devices and troubleshoot doc for tensor synchronization issue

#9258 opened 2025-05-28 21:12 by aws-yyjau

Migrate `torch_xla.device()` to `torch.device('xla')`

#9253 opened 2025-05-27 22:03 by ghpvnist

[torchax]: JittableModule statedict handling

#9195 opened 2025-05-19 10:28 by zmelumian972

Migrate to correct logger interface

#9191 opened 2025-05-17 09:32 by emmanuel-ferdman

Switch bazel to default to Clang for C/C++ compilation

#9169 opened 2025-05-14 18:26 by zhanyong-wan

[Test] Add torch pin 152601

#9167 opened 2025-05-14 18:08 by bhavya01

Test torchprime from PyTorch/XLA

#9152 opened 2025-05-13 00:48 by tengyifei

(Do not submit) Set StableHLO Compilation to true for CI

#9141 opened 2025-05-12 09:33 by GleasonK

remove version number from nightly build references

#9130 opened 2025-05-09 17:11 by bfolie

Lower clamp

#9123 opened 2025-05-08 23:25 by ghpvnist

Update numerical verification for SPMD Linear checkpointing CI

#9113 opened 2025-05-07 23:53 by sdasgup3

Collect coverage data. DO_NOT_MERGE DO_NOT_REVIEW_YET

#9107 opened 2025-05-07 17:43 by ysiraichi

Move requires_jax to inner flash_attention functions

#9098 opened 2025-05-05 23:47 by bhavya01

Add features needed for vllm

#9092 opened 2025-05-05 18:11 by wenxindongwork

[not for review] update pin

#9087 opened 2025-05-04 13:25 by bobrenjc93

Check the scan op input for requires_grad

#9083 opened 2025-05-02 20:33 by haifeng-jin

Assert on empty PJRT buffers

#9062 opened 2025-04-29 23:58 by rpsilva-aws

(Draft - PT27) Extend device data node binding to not override tensor IDs

#9052 opened 2025-04-28 23:47 by rpsilva-aws

reduce redundant graph in collective op distributed

#9044 opened 2025-04-25 22:24 by zpcore

[DRAFT] call_jax sets the JAX contextual mesh

#9043 opened 2025-04-25 19:03 by tengyifei