Skip to content

Pull requests: pytorch/xla

New pull request New
142 Open 5,735 Closed
142 Open 5,735 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

Introduce multi-node SPMD initialization for Neuron
#8046 opened Sep 19, 2024 by rpsilva-aws • Draft
Fix negative dims in scatter_index
#8045 opened Sep 19, 2024 by dvhg Loading…
[torch_xla2] Basic DDP example in a notebook
#8042 opened Sep 18, 2024 by will-cromar Loading…
parallel_loader: Fix TPU memory leak when calling __iter__.
#8039 opened Sep 18, 2024 by dudulightricks Loading…
3
add resize_as_
#8036 opened Sep 17, 2024 by ManfeiBai • Draft
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…
2
Add support for native_layer_norm
#8031 opened Sep 17, 2024 by matinehAkhlaghinia Loading…
1
Add stablediffusion
#8027 opened Sep 17, 2024 by qihqi • Draft
[benchmarks] Add docs file.
#8023 opened Sep 16, 2024 by ysiraichi • Draft
4
Enable cross entropy loss for xla autocast with FP32 precision
#7992 opened Sep 11, 2024 by avizon-aws Loading…
Creates a new table of contents to organize content
#7988 opened Sep 10, 2024 by mikegre-google Loading…
13
Fix nprocs description for xmp.spawn; error if nprocs != 1 or None
#7971 opened Sep 6, 2024 by jeffhataws Loading…
2
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…
26
[developer] Add a script to automate building torch_xla
#7876 opened Aug 18, 2024 by tengyifei Loading…
2
[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
[WIP] Use privateuseone dispatch key
#7705 opened Jul 17, 2024 by will-cromar Loading…
1
[torch_xla2] Attempt to get DDP working with torch.compile DO_NOT_REVIEW_YET
#7651 opened Jul 9, 2024 by will-cromar • Draft
ProTip! Updated in the last three days: updated:>2024-09-16.