Skip to content

Pull requests: pytorch/xla

New pull request New
188 Open 6,404 Closed
188 Open 6,404 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Support python 3.12
#9274 opened Jun 1, 2025 by zhanyong-wan • Draft
Update OpenXLA to 2025/5/30.
#9271 opened May 30, 2025 by zhanyong-wan Loading…
doc: update pytorch-on-xla-devices and troubleshoot doc for tensor synchronization issue
#9258 opened May 28, 2025 by aws-yyjau Loading…
29
Migrate torch_xla.device() to torch.device('xla')
#9253 opened May 27, 2025 by ghpvnist • Draft
1
@ghpvnist
Rollback #9146 to fix breakage
#9245 opened May 23, 2025 by ghpvnist • Draft
[torchax] Fixes test for kthvalue pytorch/xla#7458
#9223 opened May 21, 2025 by qianminj123 Loading…
Disable AMP by default on CPU
#9218 opened May 20, 2025 by haifeng-jin Loading…
1
@haifeng-jin
6
[torchax] Make copy_ works on different device.
#9211 opened May 20, 2025 by qihqi Loading…
1
3
Introduce annotate_custom_sharding binding
#9203 opened May 19, 2025 by rpsilva-aws Loading…
12
Migrate runtime.xla_device in favor of core.xla_model.xla_device
#9200 opened May 19, 2025 by ghpvnist Loading…
1
@ghpvnist
18
[torchax]: JittableModule statedict handling
#9195 opened May 19, 2025 by zmelumian972 Loading…
9
Migrate to correct logger interface
#9191 opened May 17, 2025 by emmanuel-ferdman Loading…
2
configurable number of thread for dcp
#9188 opened May 16, 2025 by aws-zhenguo Loading…
@jeffhataws
1
Refactor GetNumDevices and create GetNumGlobalDevices distributed SPMD and other distributed things.
#9184 opened May 16, 2025 by pgmoka Loading…
1
@pgmoka
9
Add interop with flax (Part 1)
#9176 opened May 15, 2025 by qihqi Loading…
2
Switch bazel to default to Clang for C/C++ compilation
#9169 opened May 14, 2025 by zhanyong-wan • Draft
1
[Test] Add torch pin 152601
#9167 opened May 14, 2025 by bhavya01 • Draft
Add cache to value_and_grad_partitioned
#9163 opened May 14, 2025 by iwknow Loading…
4
Test torchprime from PyTorch/XLA
#9152 opened May 13, 2025 by tengyifei Loading…
1
8
(Do not submit) Set StableHLO Compilation to true for CI
#9141 opened May 12, 2025 by GleasonK • Draft
remove version number from nightly build references
#9130 opened May 9, 2025 by bfolie Loading…
4
Lower clamp
#9123 opened May 8, 2025 by ghpvnist • Draft
Update numerical verification for SPMD Linear checkpointing CI CI related change
#9113 opened May 7, 2025 by sdasgup3 Loading…
1
Collect coverage data. DO_NOT_MERGE Not for merging. DO_NOT_REVIEW_YET
#9107 opened May 7, 2025 by ysiraichi • Draft
Move requires_jax to inner flash_attention functions
#9098 opened May 5, 2025 by bhavya01 Loading…
1
1
ProTip! Adding no:label will show everything without a label.