Skip to content

Pull requests: pytorch/xla

New pull request New
195 Open 6,389 Closed
195 Open 6,389 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

doc: update pytorch-on-xla-devices and troubleshoot doc for tensor synchronization issue
#9258 opened May 28, 2025 by aws-yyjau Loading…
19
Run test_collective_permute.py on TPU CI.
#9257 opened May 28, 2025 by ysiraichi Loading…
Migrate torch_xla.device() to torch.device('xla')
#9253 opened May 27, 2025 by ghpvnist • Draft
1
@ghpvnist
[Kernel] support kv cache quantization in ragged attention kernel
#9249 opened May 27, 2025 by yaochengji Loading…
13
Rollback #9146 to fix breakage
#9245 opened May 23, 2025 by ghpvnist • Draft
Update README.md with supported Python version
#9239 opened May 22, 2025 by zzzwen Loading…
@zzzwen
5
[torchax] Fixes test for kthvalue pytorch/xla#7458
#9223 opened May 21, 2025 by qianminj123 Loading…
[torchax] Added support for bicubic and billinear resampling
#9222 opened May 21, 2025 by qihqi Loading…
3
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…
10
Migrate runtime.xla_device in favor of core.xla_model.xla_device
#9200 opened May 19, 2025 by ghpvnist Loading…
1
@ghpvnist
18
[torchax] Added support for bicubic and billinear resampling
#9196 opened May 19, 2025 by zmelumian972 Loading…
6
[torchax]: JittableModule statedict handling
#9195 opened May 19, 2025 by zmelumian972 Loading…
9
Improve torch_xla.compile documentation
#9194 opened May 19, 2025 by sdasgup3 Loading…
1
27
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
6
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…
3
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
In pjrt runtime client, raise a Python exception if XLA compilation fails.
#9138 opened May 10, 2025 by zhanyong-wan • Draft
5
ProTip! Type g i on any issue or pull request to go back to the issue listing page.