Skip to content

Pull requests: pytorch/torchtitan

Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
Reviews
Assignee
Filter by who’s assigned
Sort

Pull requests list

Update requirements.txt
#323 opened May 11, 2024 by qiziAI Loading…
Make Transformer tolerate missing layers for PP CLA Signed This label is managed by the Meta Open Source bot.
#322 opened May 10, 2024 by wconstab Loading…
Refactor freqs_cis slice to be safer for PP CLA Signed This label is managed by the Meta Open Source bot.
#321 opened May 10, 2024 by wconstab Loading…
selective compilation - norm layers only CLA Signed This label is managed by the Meta Open Source bot.
#320 opened May 10, 2024 by lessw2020 Loading…
Add support of DDP and CompiledAutograd. CLA Signed This label is managed by the Meta Open Source bot.
#319 opened May 9, 2024 by fegin Loading…
Add Pipeline Parallel (and 2D PP+FSDP) support CLA Signed This label is managed by the Meta Open Source bot.
#318 opened May 9, 2024 by wconstab Loading…
[fused_rmsnorm] Register as a custom operator for tracing CLA Signed This label is managed by the Meta Open Source bot.
#303 opened May 3, 2024 by wconstab Loading…
[fused_rmsnorm] Avoid querying device inside forward CLA Signed This label is managed by the Meta Open Source bot.
#301 opened May 3, 2024 by wconstab Loading…
[fused_rmsnorm] Avoid conditional on dynamic stride CLA Signed This label is managed by the Meta Open Source bot.
#300 opened May 3, 2024 by wconstab Loading…
register fused rmsnorm as pytorch custom op CLA Signed This label is managed by the Meta Open Source bot.
#296 opened May 2, 2024 by tianyu-l Draft
[wip] differentiate Rstd vs rstd CLA Signed This label is managed by the Meta Open Source bot.
#294 opened May 2, 2024 by lessw2020 Loading…
Enable TP+PP support CLA Signed This label is managed by the Meta Open Source bot.
#285 opened Apr 29, 2024 by wconstab Loading…
Use stateful dataloader to checkpoint data iteration order and token buffer CLA Signed This label is managed by the Meta Open Source bot.
#279 opened Apr 26, 2024 by gokulavasan Loading…
torch.compile each TransformerBlock instead of the whole model CLA Signed This label is managed by the Meta Open Source bot.
#268 opened Apr 25, 2024 by wanchaol Loading…
RFC for ckpt apis CLA Signed This label is managed by the Meta Open Source bot.
#226 opened Apr 13, 2024 by wconstab Loading…
[RFC] Sharded embeddings in separate FSDP group CLA Signed This label is managed by the Meta Open Source bot.
#182 opened Apr 1, 2024 by awgu Draft
run sdpa with dtensor CLA Signed This label is managed by the Meta Open Source bot.
#180 opened Mar 30, 2024 by tianyu-l Loading…
[Draft] PP tracing works CLA Signed This label is managed by the Meta Open Source bot.
#155 opened Mar 20, 2024 by kwen2501 Draft
Implement fast checkpoint path CLA Signed This label is managed by the Meta Open Source bot.
#127 opened Mar 12, 2024 by fegin Loading…
ProTip! Add no:assignee to see everything that’s not assigned.