Skip to content

Pull requests: pytorch/xla

New pull request New
139 Open 5,773 Closed
139 Open 5,773 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

[benchmarks] Default to functionalization disabled.
#8093 opened Sep 28, 2024 by ysiraichi Loading…
Fix torch.min and torch.mode WIP (#7513)
#8092 opened Sep 28, 2024 by simonteozw Loading…
Implement torch.ops.aten.embedding_renorm_
#8091 opened Sep 28, 2024 by guyao Loading…
1
[WIP] Implement upsample_bilinear2d_aa
#8090 opened Sep 27, 2024 by barney-s Loading…
1
Pin update
#8079 opened Sep 26, 2024 by qihqi Loading…
add argwhere
#8073 opened Sep 25, 2024 by ManfeiBai • Draft
1
Check warning msg in second captured warning.
#8069 opened Sep 25, 2024 by zhuhaozhe Loading…
change r2.5 to use jax stable version 0.4.33 2.5 release tpuci
#8062 opened Sep 24, 2024 by ManfeiBai Loading…
Support FSDPv2 compute dtype
#8056 opened Sep 24, 2024 by lausannel Loading…
4
Print the training update after the device execution DO_NOT_REVIEW_YET
#8054 opened Sep 24, 2024 by JackCaoG • Draft
Introduce multi-node SPMD initialization for Neuron
#8046 opened Sep 19, 2024 by rpsilva-aws • Draft
2
Add profiler annotation for the decoderonly example tpuci
#8034 opened Sep 17, 2024 by JackCaoG • Draft
1
Forward XLATensorImpl::is_contiguous_custom to TensorImpl.
#8032 opened Sep 17, 2024 by ysiraichi Loading…
12
Add support for native_layer_norm
#8031 opened Sep 17, 2024 by matinehAkhlaghinia Loading…
1
[benchmarks] Add docs file.
#8023 opened Sep 16, 2024 by ysiraichi Loading…
7
Creates a new table of contents to organize content
#7988 opened Sep 10, 2024 by mikegre-google Loading…
13
support dist.broadcast tpuci usability Bugs/features related to improving the usability of PyTorch/XLA
#7956 opened Sep 5, 2024 by zpcore Loading…
6
Update dedup logic to use dedup_save_plans
#7929 opened Aug 29, 2024 by jonb377 Loading…
4
Update comments in data parallel example to use sampler
#7914 opened Aug 26, 2024 by JackCaoG Loading…
1
6
scan and apply_layers
#7901 opened Aug 23, 2024 by tengyifei Loading…
@tengyifei
56
[WIP] Skip AOTAutograd view-reconstruction for inference runs. xla:gpu
#7834 opened Aug 12, 2024 by ysiraichi • Draft
1
Initial pr to support dynamic shape detection
#7817 opened Aug 7, 2024 by JackCaoG • Draft
1
API_GUIDE.md: fix execution => exception typo + rewording
#7808 opened Aug 6, 2024 by houeland Loading…
Support SplashAttention in PyTorch/XLA tpuci
#7798 opened Aug 1, 2024 by wonjoolee95 Loading…
2
usability
#7728 opened Jul 23, 2024 by zpcore • Draft
ProTip! Adding no:label will show everything without a label.