Skip to content

Pull requests: pytorch/xla

New pull request New
159 Open 6,235 Closed
159 Open 6,235 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

Add test to ensure scan-based and the standard GRU are interchangeable.
#8949 opened Apr 7, 2025 by iwknow Loading…
2
Update test_export_fx_passes.py
#8933 opened Apr 3, 2025 by avikchaudhuri Loading…
add CreateGlobalShardedData prototype
#8932 opened Apr 3, 2025 by pgmoka • Draft
1
@pgmoka
test on debian12
#8928 opened Apr 2, 2025 by zpcore • Draft
8
@assume_pure
#8923 opened Apr 2, 2025 by tengyifei • Draft
Make GPU CUDA plugin require JAX
#8919 opened Apr 1, 2025 by tengyifei • Draft
Lower upsample_bilinear2d_backward operation.
#8916 opened Apr 1, 2025 by ysiraichi • Draft
1
1
Adapt Splash Attention from TorchPrime
#8911 opened Mar 31, 2025 by zpcore Loading…
21
[Draft] Add Experimental limited sparse embedding bag
#8905 opened Mar 28, 2025 by amjames • Draft
1
@lsy323
[ragged-paged-attn] Use hidden states in kv cache and support any num…
#8891 opened Mar 26, 2025 by zpcore Loading…
[DRAFT/WIP] Add top-p masking
#8871 opened Mar 21, 2025 by hyeygit • Draft
[1/N] Initial implementation of local SPMD support distributed SPMD and other distributed things.
#8810 opened Mar 9, 2025 by lsy323 Loading…
@lsy323
39
Showcase jax.grad in torch_xla
#8800 opened Mar 5, 2025 by zpcore Loading…
6
Repro ragged paged attn kernel
#8752 opened Feb 26, 2025 by vanbasten23 • Draft
Replace setup.py with pyproject.toml
#8744 opened Feb 26, 2025 by ManfeiBai Loading…
Follow up on ragged kernel wrapper
#8737 opened Feb 24, 2025 by vanbasten23 • Draft
[SPMD] Add API to disable global SPMD config
#8717 opened Feb 18, 2025 by lsy323 • Draft
9
[SPMD] Add API to create global tensor from local CPU/TPU shards
#8716 opened Feb 18, 2025 by lsy323 • Draft
Document how to debug the dispatcher
#8712 opened Feb 15, 2025 by tengyifei Loading…
4
Add instruction for exporting inlined constant
#8707 opened Feb 13, 2025 by qihqi Loading…
4
enable tensor.copy_ with torch native cpu tensor as source
#8682 opened Feb 6, 2025 by yaochengji Loading…
2
Transition to Hermetic CUDA
#8665 opened Feb 3, 2025 by ysiraichi • Draft
@ysiraichi
2
add aarch64 platform build support
#8663 opened Feb 2, 2025 by snadampal Loading…
3
Lower cummin op lowering ATen Operation lowering
#8565 opened Jan 14, 2025 by zyy-martin Loading…
1
@zyy-martin
12
Implement torch.linspace and torch.logspace
#8533 opened Jan 6, 2025 by vyom1611 Loading…
6
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.