Skip to content

Pull requests: pytorch/xla

New pull request New
146 Open 5,294 Closed
146 Open 5,294 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
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

Modularize zero step function
#7265 opened Jun 13, 2024 by karakusc Loading…
Make a dynamo inference test that runs on TPU dynamo
#7264 opened Jun 13, 2024 by JackCaoG Loading…
Increase max in flight operation to accommodate eager mode
#7263 opened Jun 12, 2024 by JackCaoG Loading…
1
Backport: Bump jaxversion corresponding to libtpu 0612
#7262 opened Jun 12, 2024 by vanbasten23 Loading…
[benchmarks] Initialize moco and TPU experiments on XLA. xla:gpu
#7257 opened Jun 12, 2024 by ysiraichi Loading…
1
2
support eager mode with torch.compile
#7256 opened Jun 12, 2024 by JackCaoG Loading…
[ZeRO-1] Only warn on rank-0
#7252 opened Jun 12, 2024 by hgt312 Loading…
Lower upsample_nearest_3d and its backward
#7251 opened Jun 12, 2024 by wonjoolee95 • Draft
[SPMD] Fix reduce_scatter for TPUv2 backport_2.4 tpuci
#7248 opened Jun 11, 2024 by alanwaketan Loading…
@alanwaketan
4
Update cuda_deps.yaml to include CUDA 12.4 backport_2.4
#7244 opened Jun 11, 2024 by bhavya01 Loading…
@bhavya01
2
Sample training UX
#7220 opened Jun 7, 2024 by qihqi • Draft
2
Bucketing gradients
#7216 opened Jun 7, 2024 by amithrm Loading…
13
Add a test to convert xla gpu tensor to cuda tensor by using the protocol.
#7213 opened Jun 6, 2024 by vanbasten23 Loading…
Updated the examples folder readme file
#7208 opened Jun 6, 2024 by sitamgithub-MSIT Loading…
1
Integrate dlpack to dynamo.
#7173 opened Jun 3, 2024 by vanbasten23 Loading…
17
[torch_xla2] Add lowerings for rand_like, randn_like, and uniform
#7170 opened Jun 3, 2024 by will-cromar • Draft
[Fori_loop|While_loop] Enable while_loop/fori_loop, add test case #7094
#7157 opened May 30, 2024 by ManfeiBai Loading…
49
Add examples to show case llama training.
#7129 opened May 28, 2024 by qihqi Loading…
Add jlibrary module to torch_xla2 for preserving library abstractions during export
#7128 opened May 28, 2024 by GleasonK Loading…
@qihqi @lsy323
[OpenXLA-pin-update trigger] Create _openxla_pin_update_weekly.yml
#7108 opened May 24, 2024 by ManfeiBai Loading…
26
docs: fix typos in spmd.md
#7065 opened May 15, 2024 by clumsy Loading…
Run persistent cache test on TPU CI tpuci
#7038 opened May 8, 2024 by jonb377 Loading…
3
Bump jinja2 from 3.1.3 to 3.1.4 in /docs dependencies Pull requests that update a dependency file
#7027 opened May 6, 2024 by dependabot bot Loading…
add check for neuron root comm id if not set
#7007 opened Apr 30, 2024 by aws-kingrj Loading…
3
Support ahead of time (AOT) cross-device compilation AOT ahead of time cross device compilation
#6992 opened Apr 29, 2024 by zpcore Loading…
5
ProTip! no:milestone will show everything without a milestone.