Skip to content

Pull requests: pytorch/xla

New pull request New
196 Open 6,376 Closed
196 Open 6,376 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Modify NoneRemover to return modified list/tuple
#9232 opened May 21, 2025 by ghpvnist • Draft
@ghpvnist
1
Separate pallas test into a separate test shard
#9231 opened May 21, 2025 by ghpvnist • Draft
Add description of prerequisites for running benchmarks
#9230 opened May 21, 2025 by haifeng-jin Loading…
1
@haifeng-jin
1
Ci
#9229 opened May 21, 2025 by ghpvnist • Draft
Revert "Ensure that all of PyTorch/XLA C++ is compiled with exception… ci-no-td
#9225 opened May 21, 2025 by yaoshiang Loading…
1
@yaoshiang
[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…
1
Disable AMP by default on CPU
#9218 opened May 20, 2025 by haifeng-jin Loading…
1
@haifeng-jin
2
Revert "Switch bazel default spawn strategy to sandboxed" ci-no-td
#9212 opened May 20, 2025 by zhanyong-wan • Draft
[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…
1
Migrate runtime.xla_device in favor of core.xla_model.xla_device
#9200 opened May 19, 2025 by ghpvnist • Draft
1
@ghpvnist
4
[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…
8
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…
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…
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! Follow long discussions with comments:>50.