Skip to content

Pull requests: pytorch/xla

New pull request New
188 Open 6,757 Closed
188 Open 6,757 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

Fix persistent compilation cache not working on TPU
#9759 opened Mar 23, 2026 by shr1ram Loading…
3 tasks
1
Fix scan_layers to support layers with no weights
#9758 opened Mar 21, 2026 by kredd2506 Loading…
2 tasks
1
Make a function wrapper for _get_xla_sharding_specs
#9757 opened Mar 17, 2026 by ajakovljevicTT Loading…
1
Optimize as_strided_copy fast path to support offset
#9728 opened Dec 10, 2025 by Zantares Loading…
Expose torch_xla.runtime.clear_computation_cache binding to clear computation cache
#9720 opened Nov 24, 2025 by jameszianxuTT Loading…
5
Fix typo in warning message for SPMD mode
#9718 opened Nov 20, 2025 by ddilbazTT Loading…
implement XLAHooks and register it to PyTorch when loaded.
#9683 opened Oct 20, 2025 by iwknow Loading…
1
4
Ensuring PJRT Client destroy/destructor is called
#9675 opened Oct 7, 2025 by saarthak-aws Loading…
13
Add destructor for XLAGraphExecutor to free resource
#9636 opened Sep 11, 2025 by Zantares Loading…
2
Test ci
#9633 opened Sep 10, 2025 by bhavya01 Loading…
Adding functionality for metadata communication across hosts
#9570 opened Aug 19, 2025 by kvshbg-aws Loading…
8
xla: relax error message assertion in dot_general test (allow backend…
#9557 opened Aug 13, 2025 by Rahulchaube1 Loading…
2
Added Multi-Host TPU tutorial
#9507 opened Jul 24, 2025 by vfdev-5 Loading…
25
XLAShardedTensor.to_local() support
#9505 opened Jul 24, 2025 by Hoomaaan Loading…
10
Add tests for DTensor.from_local() support for XLA device mesh
#9504 opened Jul 24, 2025 by Hoomaaan Loading…
Support SPMD placeholder tensors
#9489 opened Jul 17, 2025 by rpsilva-aws • Draft
Update torchprime tests to Python 3.12 bug Something isn't working testing Testing and coverage related issues.
#9481 opened Jul 15, 2025 by pgmoka • Draft
@pgmoka
feat: abstraction of xla::OpSharding proto using wrapper class
#9467 opened Jul 10, 2025 by kvshbg-aws Loading…
25
Introduce multi-operand collective permute
#9450 opened Jul 8, 2025 by rpsilva-aws Loading…
Convert some XLA_CHECKs to fatal errors.
#9439 opened Jul 2, 2025 by zhanyong-wan • Draft
Add a test on OOM error
#9414 opened Jun 26, 2025 by zhanyong-wan • Draft
[wip]ruff inter on torchax
#9396 opened Jun 20, 2025 by zzzwen Loading…
1
[WIP] Add autocast in torchax
#9361 opened Jun 16, 2025 by qihqi • Draft
Clarify xr.device_type() API and use them
#9343 opened Jun 11, 2025 by ghpvnist Loading…
@ghpvnist
wip: use torch from a wheel
#9340 opened Jun 11, 2025 by rickeylev • Draft
1 task
5
ProTip! Follow long discussions with comments:>50.