Skip to content

Pull requests: Lightning-AI/lightning-thunder

New pull request New
39 Open 1,060 Closed
39 Open 1,060 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

numpy 2.0 dependencies
#1748 opened Feb 6, 2025 by t-vi Loading…
[Thunderfx report] Adds Thunder segmentation and fusion information
#1747 opened Feb 5, 2025 by kiya00 • Draft
4 tasks
1
6
Adds a benchmark of MoE layer with deepseek-ai/DeepSeek-R1 config
#1746 opened Feb 5, 2025 by kiya00 Loading…
4 tasks
12
type annotation minor change: [Trace] -> list[Trace]
#1742 opened Feb 4, 2025 by crcrpar Loading…
fixing TensorProxy.__init__(like=a) to accept a.requires_grad
#1725 opened Jan 31, 2025 by jjsjann123 Loading…
[WIP] TE: update for TE release_v2.0 support TransformerEngine
#1721 opened Jan 30, 2025 by kshitij12345 • Draft
1
Add HF benchmarks benchmarking hf-transformers huggingface For supporting HF models peft
#1720 opened Jan 30, 2025 by riccardofelluga Loading…
1
4
Add benchmark isolation option benchmarking documentation Improvements or additions to documentation
#1719 opened Jan 30, 2025 by riccardofelluga Loading…
Relax test_te_with_retain_graph xfail strictness in test_transformer_engine_executor.py
#1718 opened Jan 30, 2025 by nWEIdia Loading…
3 tasks
FIx typos in adding_custom_operator.ipynb
#1717 opened Jan 29, 2025 by qqaatw Loading…
4 tasks
1
Grad transform patch
#1712 opened Jan 29, 2025 by jjsjann123 • Draft
10
[DO NOT REVIEW][SMOKE TEST] Skip prims shape in grad transform
#1704 opened Jan 27, 2025 by jjsjann123 • Draft
5
[WIP] Test out thunder.jit w/ NeMo models.
#1694 opened Jan 24, 2025 by tfogal • Draft
3
bump litgpt and bitsandbytes dependencies
#1691 opened Jan 24, 2025 by t-vi • Draft
fix TE distributed tests with fsdp_v2 and ddp_v2
#1690 opened Jan 24, 2025 by kshitij12345 Loading…
2
Remove prims.embedding and prims.embedding_backward operators
#1689 opened Jan 24, 2025 by IvanYashchuk Loading…
10
make no_sync behave like in pytorch
#1681 opened Jan 22, 2025 by t-vi • Draft
1
Add torchcompile_xentropy executor
#1655 opened Jan 17, 2025 by riccardofelluga Loading…
1
4
[WIP] initial changes to accommodate pipeline parallelism transform
#1650 opened Jan 16, 2025 by ali-alshaar7 • Draft
4 tasks
revise examine memory's collection handling, output memory annotation in benchmark_litgpt
#1634 opened Jan 10, 2025 by t-vi • Draft
Remove requires_grad from regular TensorProxy and add specialized TensorProxy with requires_grad and is_leaf attributes ci dependencies developer efficiency documentation Improvements or additions to documentation install
#1599 opened Dec 30, 2024 by IvanYashchuk • Draft
8
[tensor subclass] print type_string of tensor attributes
#1592 opened Dec 29, 2024 by crcrpar Loading…
1
check tensor attrs of tensor wrapper subclasses in prologue
#1591 opened Dec 28, 2024 by crcrpar Loading…
[tensor wrapper subclass] Add support for torchao.float8 mlp
#1585 opened Dec 23, 2024 by crcrpar • Draft
1
[tensor wrapper subclass] Add trace transform for tensor subclasses documentation Improvements or additions to documentation
#1584 opened Dec 23, 2024 by crcrpar • Draft
ProTip! Type g i on any issue or pull request to go back to the issue listing page.