Skip to content

Pull requests: pytorch/xla

New pull request New
151 Open 5,931 Closed
151 Open 5,931 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

flash_attention: support also cross attention. tpuci
#8427 opened Nov 28, 2024 by dudulightricks Loading…
1
[torch_xla2] Add transposed conv ops
#8426 opened Nov 28, 2024 by dvhg Loading…
1
Reenable the distributed checkpointing test tpuci
#8424 opened Nov 27, 2024 by JackCaoG Loading…
Add autocast support for XlaPatchedLinear
#8421 opened Nov 27, 2024 by aws-nm9 Loading…
6
Add autocast support for einsum
#8420 opened Nov 27, 2024 by aws-nm9 Loading…
5
[scan] Add a test under SPMD tpuci
#8419 opened Nov 26, 2024 by tengyifei Loading…
5
[LoweringContext] SPMD support
#8418 opened Nov 26, 2024 by rpsilva-aws Loading…
2
update OpenXLA-pin to 11/26
#8417 opened Nov 26, 2024 by ManfeiBai • Draft
[Documentation] Added a section pointing readers to the AI-Hypercomputer/tpu-recipies repo for reference model implementations.
#8412 opened Nov 25, 2024 by mikegre-google Loading…
Switch to libtpu package from the libtpu-wheels registry tpuci
#8409 opened Nov 22, 2024 by tengyifei Loading…
2
[torch_xla2] Update test_ops.py
#8403 opened Nov 20, 2024 by ManfeiBai Loading…
Migrate from capture_pre_autograd_graph to export_for_training
#8398 opened Nov 19, 2024 by yushangdi Loading…
1
[torch_xla2] Create jtorchvision_roi_align.py
#8394 opened Nov 19, 2024 by ManfeiBai • Draft
norm and pca_lowrank op info test from issue #7528
#8384 opened Nov 14, 2024 by vyom1611 Loading…
Align rrelu_with_noise schema to reflect noise mutation [branch]
#8363 opened Nov 7, 2024 by IvanKobzarev Loading…
WIP implement lowering for torchvision::roi_align
#8362 opened Nov 7, 2024 by barney-s Loading…
1
DLPack: fix test using PyTorch API.
#8348 opened Nov 1, 2024 by ysiraichi Loading…
1
Updates Contributing.md documentation
#8343 opened Oct 30, 2024 by mikegre-google Loading…
@miladm
custom_kernel: fix shape mismatch by sharding segment_ids in flash attn. pallas
#8333 opened Oct 29, 2024 by dudulightricks • Draft
4
[release] add permission of iam.auto.tfvars
#8326 opened Oct 25, 2024 by ManfeiBai Loading…
2
print payload DO_NOT_MERGE Not for merging.
#8312 opened Oct 24, 2024 by zpcore • Draft
Align rrelu_with_noise schema to reflect noise mutation
#8309 opened Oct 24, 2024 by IvanKobzarev Loading…
4
debug DO_NOT_MERGE Not for merging.
#8293 opened Oct 21, 2024 by zpcore • Draft
[DO NOT REVIEW] Experiment the pl.debug_print in Pallas.
#8284 opened Oct 18, 2024 by vanbasten23 • Draft
[DO NOT REVIEW YET] Extend paged attention
#8237 opened Oct 8, 2024 by vanbasten23 • Draft
ProTip! Mix and match filters to narrow down what you’re looking for.