Skip to content

Pull requests: pytorch/xla

New pull request New
189 Open 6,487 Closed
189 Open 6,487 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

Pins update for week of 6/16/2025
#9388 opened Jun 18, 2025 by yaoshiang • Draft
1
Add Status and StatusOr<T> QOL functions.
#9386 opened Jun 18, 2025 by ysiraichi • Draft
1
Fix nested stableHLO composite regions
#9385 opened Jun 18, 2025 by Carlomus Loading…
ErrorHandling: make GetComputationClient() return StatusOr<T> type.
#9384 opened Jun 18, 2025 by ysiraichi Loading…
23
Add jax_device context manager to control the device target
#9382 opened Jun 17, 2025 by zzzwen Loading…
2
Torchax: Allow reuse of the jittable procedure with different functio…
#9374 opened Jun 17, 2025 by zmelumian972 Loading…
implement send and recv using collective_permute
#9373 opened Jun 17, 2025 by bfolie Loading…
15
Support torch.distributed.scatter collective
#9365 opened Jun 16, 2025 by bfolie Loading…
21
[WIP] Add autocast in torchax
#9361 opened Jun 16, 2025 by qihqi • Draft
Refactor call_jax to allow implementing ops in python
#9354 opened Jun 13, 2025 by qihqi Loading…
7
s/torch_xla2/torchax
#9353 opened Jun 13, 2025 by tengyifei Loading…
1
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
Test making this work from forks
#9335 opened Jun 11, 2025 by tengyifei • Draft
Do not inherit secrets
#9331 opened Jun 10, 2025 by tengyifei • Draft
Update tuned block table and add benchmarking/autotune scripts for quantized matmul pallas kernel.
#9322 opened Jun 10, 2025 by vanbasten23 • Draft
Add DLRM DCN v2 with JAX Sparsecore
#9287 opened Jun 3, 2025 by chandrasekhard2 Loading…
1
WIP - do not review; Support python 3.12
#9274 opened Jun 1, 2025 by zhanyong-wan • Draft
1
Migrate torch_xla.device() to torch.device('xla')
#9253 opened May 27, 2025 by ghpvnist • Draft
1
@ghpvnist
6
[torchax]: JittableModule statedict handling
#9195 opened May 19, 2025 by zmelumian972 Loading…
10
Migrate to correct logger interface
#9191 opened May 17, 2025 by emmanuel-ferdman 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
(Do not submit) Set StableHLO Compilation to true for CI
#9141 opened May 12, 2025 by GleasonK • Draft
Lower clamp
#9123 opened May 8, 2025 by ghpvnist • Draft
ProTip! no:milestone will show everything without a milestone.