Skip to content

Pull requests: pytorch/xla

New pull request New
190 Open 6,627 Closed
190 Open 6,627 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

Update XLA pin then fix up to make it compile
#9565 opened Aug 15, 2025 by qihqi Loading…
full: improve error handling and error messages.
#9564 opened Aug 15, 2025 by ysiraichi Loading…
add CI for jax 0.7.1
#9563 opened Aug 14, 2025 by qihqi Loading…
Rename MaybeThrow to OkOrThrow.
#9561 opened Aug 14, 2025 by ysiraichi Loading…
1
xla: relax error message assertion in dot_general test (allow backend…
#9557 opened Aug 13, 2025 by Rahulchaube1 Loading…
1
[EZ] Replace pytorch-labs with meta-pytorch
#9556 opened Aug 12, 2025 by ZainRizvi Loading…
1
Add Shardy support for Torch-XLA
#9541 opened Aug 5, 2025 by hshahTT Loading…
5
Add xla random generator.
#9539 opened Aug 5, 2025 by iwknow Loading…
8
Generate docstrings using gemini.
#9538 opened Aug 4, 2025 by qihqi Loading…
Make torch_xla package PEP 561 compliant
#9515 opened Jul 28, 2025 by wirthual Loading…
2
feat: infer device_ids and normalize tile assignment
#9514 opened Jul 26, 2025 by kvshbg-aws Loading…
Added Multi-Host TPU tutorial
#9507 opened Jul 24, 2025 by vfdev-5 • Draft
22
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…
Add Quickstart & Migration guide
#9503 opened Jul 23, 2025 by pavithraes Loading…
7
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…
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
implement send and recv using collective_permute
#9373 opened Jun 17, 2025 by bfolie Loading…
20
[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
ProTip! Add no:assignee to see everything that’s not assigned.