Skip to content

Pull requests: pytorch/xla

New pull request New
186 Open 6,323 Closed
186 Open 6,323 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

Silence distributed warning
#9140 opened May 10, 2025 by tengyifei Loading…
Test torchax on Python 3.10 - 3.12
#9139 opened May 10, 2025 by tengyifei Loading…
Throw a Python exception if compilation fails.
#9138 opened May 10, 2025 by zhanyong-wan • Draft
Add assume_pure_torch implementation for forward pass only
#9135 opened May 9, 2025 by bhavya01 • Draft
2
Document the difference between tracing time and execution time documentation
#9133 opened May 9, 2025 by sdasgup3 Loading…
1
7
remove version number from nightly build references
#9130 opened May 9, 2025 by bfolie Loading…
3
Yifeit/torchax 2 7
#9126 opened May 9, 2025 by tengyifei • Draft
Lower clamp
#9123 opened May 8, 2025 by ghpvnist • Draft
get master ip address for neuron device
#9120 opened May 8, 2025 by aws-zhenguo Loading…
[benchmarks] Fix run single config command error documentation
#9115 opened May 8, 2025 by haifeng-jin Loading…
@haifeng-jin
1
Update numerical verification for SPMD Linear checkpointing CI CI related change
#9113 opened May 7, 2025 by sdasgup3 Loading…
1
Collect coverage data. DO_NOT_MERGE Not for merging. DO_NOT_REVIEW_YET
#9107 opened May 7, 2025 by ysiraichi • Draft
9082 educate users on mat mul precision
#9103 opened May 6, 2025 by yaoshiang Loading…
41
Move requires_jax to inner flash_attention functions
#9098 opened May 5, 2025 by bhavya01 Loading…
1
1
Add features needed for vllm
#9092 opened May 5, 2025 by wenxindongwork • Draft
[not for review] update pin
#9087 opened May 4, 2025 by bobrenjc93 Loading…
3
Replace implementation of xla_torch.sync() to xm.mark_step()
#9086 opened May 3, 2025 by ghpvnist Loading…
9
Check the scan op input for requires_grad
#9083 opened May 2, 2025 by haifeng-jin Loading…
1
@haifeng-jin
1
Refine the gradient accumulation API
#9078 opened May 1, 2025 by rpsilva-aws Loading…
8
[cherry-pick to r2.6_aws_neuron] add torch_xla_graph_execution_check_level (default disabled) flag that emits warning(1) or throw error(2) during tensor sync and output the python frame
#9077 opened May 1, 2025 by aws-yyjau Loading…
1
Fix + Run DynamicShapeDetector tests on CI.
#9075 opened May 1, 2025 by ysiraichi Loading…
Disable running slow CI for doc-only changes
#9072 opened May 1, 2025 by ghpvnist Loading…
1
@ghpvnist
13
Replace xm.mark_step with torch_xla.sync() wherever possible
#9070 opened May 1, 2025 by ghpvnist Loading…
1
5
Assert on empty PJRT buffers
#9062 opened Apr 29, 2025 by rpsilva-aws Loading…
4
add torch_xla_graph_execution_check_level (default disabled) flag that emits warning(1) or throw error(2) during tensor sync and output the python frame
#9057 opened Apr 29, 2025 by aws-yyjau Loading…
3
ProTip! no:milestone will show everything without a milestone.