Skip to content

Pull requests: pytorch/xla

New pull request New
197 Open 6,366 Closed
197 Open 6,366 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

Fixed reloading of sharded master weights
#9224 opened May 21, 2025 by avizon-aws Loading…
[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…
Revert "Fix a clang compilation error. (#9156)" ci-no-td
#9221 opened May 21, 2025 by yaoshiang Loading…
1
[ragged-paged-attn] Apply kv mask to filter out NaNs
#9219 opened May 21, 2025 by bythew3i Loading…
1
Disable AMP by default on CPU
#9218 opened May 20, 2025 by haifeng-jin Loading…
1
@haifeng-jin
Update the OpenXLA pin
#9216 opened May 20, 2025 by haifeng-jin Loading…
1
@haifeng-jin
1
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
Subset of existing, approved PR
#9204 opened May 19, 2025 by yaoshiang Loading…
1
Part 1: Introduce annotate_custom_sharding binding
#9203 opened May 19, 2025 by rpsilva-aws • Draft
Migrate runtime.xla_device in favor of core.xla_model.xla_device
#9200 opened May 19, 2025 by ghpvnist • Draft
1
@ghpvnist
[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
22
Migrate to correct logger interface
#9191 opened May 17, 2025 by emmanuel-ferdman Loading…
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
5
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
wip: test
#9152 opened May 13, 2025 by tengyifei • Draft
(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! Find all pull requests that aren't related to any open issues with -linked:issue.