Skip to content

Pull requests: pytorch/xla

New pull request New
146 Open 5,259 Closed
146 Open 5,259 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

Update gpu.md to run ResNet example on GPU
#7222 opened Jun 8, 2024 by ahoblitz Loading…
1
Modularize zero optimizer step function
#7221 opened Jun 7, 2024 by karakusc Loading…
Sample training UX
#7220 opened Jun 7, 2024 by qihqi • Draft
2
Register custom call with CUDA plugin
#7219 opened Jun 7, 2024 by bhavya01 Loading…
Bucketing gradients
#7216 opened Jun 7, 2024 by amithrm Loading…
9
Add a test to convert xla gpu tensor to cuda tensor by using the protocol.
#7213 opened Jun 6, 2024 by vanbasten23 Loading…
Updating ml_framework_name and version to pytorch/xla2 in jax_configs.
#7211 opened Jun 6, 2024 by aman2930 Loading…
4
Updated the examples folder readme file
#7208 opened Jun 6, 2024 by sitamgithub-MSIT Loading…
Integrate dlpack to dynamo.
#7173 opened Jun 3, 2024 by vanbasten23 Loading…
[torch_xla2] Add lowerings for rand_like, randn_like, and uniform
#7170 opened Jun 3, 2024 by will-cromar • Draft
add python3.11 doc for XLA v2.3
#7166 opened Jun 1, 2024 by Mon-ius Loading…
torchbench CI CI CI related change
#7162 opened May 31, 2024 by zpcore Loading…
14
[Fori_loop|While_loop] Enable while_loop/fori_loop, Add linear/MNIST test case #7094
#7157 opened May 30, 2024 by ManfeiBai Loading…
28
Use dest_offsets directly in LoadPlanner
#7155 opened May 30, 2024 by jonb377 Loading…
[ZeRO-1] Sync features from r2.1_aws_neuron branch
#7132 opened May 29, 2024 by hgt312 Loading…
1
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
Megablox tgmm kernel implementation pallas
#7032 opened May 6, 2024 by miladm Loading…
@miladm
2
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
[DO NOT REVIEW YET] Zero copy tensors between CUDA and XLA
#6970 opened Apr 25, 2024 by vanbasten23 • Draft
ProTip! Follow long discussions with comments:>50.