Skip to content

Pull requests: pytorch/torchrec

New pull request New
311 Open 2,334 Closed
311 Open 2,334 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

Test all_to_all_single CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#874 opened Dec 8, 2022 by lequytra
Create OptimizerModule for recursive state_dict and load_state_dict functionality in KeyedOptimizer CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#852 opened Dec 1, 2022 by hjmshi
3
add fsdp composability test with torchrec CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#834 opened Nov 18, 2022 by colin2328
6
Migrate gpu unittest to Pytorch test infra CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#817 opened Nov 14, 2022 by lequytra Loading…
5
Use separate pg for input dist CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#812 opened Nov 10, 2022 by xing-liu Loading…
1
fp8rowwise quantized communication in training (#1423) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#754 opened Oct 26, 2022 by xiaosun86 Loading…
3
Update _maybe_compute_kjt_to_jt_dict to support JIT tracing CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#750 opened Oct 25, 2022 by zxpmirror1994 Loading…
3
Authoring Aware ROO on LTV ReAgent model CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#742 opened Oct 24, 2022 by luofangyuyx Loading…
1
ci: Migrate unittest_ci to generic linux_job CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#739 opened Oct 21, 2022 by seemethere • Draft
Upstream apply_optim_in_backward from TorchRec CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#738 opened Oct 20, 2022 by rohan-varma Loading…
1
Allow pass in fused params in default sharders CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#734 opened Oct 18, 2022 by 842974287 Loading…
2
compute kernel derived from user CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#716 opened Oct 12, 2022 by YLGH Loading…
2
Remove test installation CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#709 opened Oct 11, 2022 by YLGH Loading…
create fx argument for sparsefeature types CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#708 opened Oct 11, 2022 by YazhiGao Loading…
1
fix state_dict compatibility with ebc family CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#703 opened Oct 10, 2022 by YazhiGao Loading…
1
Remove ShardedTensor-specific copy_ from KeyedOptimizer CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#695 opened Oct 6, 2022 by lequytra Loading…
1
support int4 quantization in prospector CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#688 opened Oct 5, 2022 by nsiddharth Loading…
1
Variable batch size RW ahd TW sequence sharding CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#687 opened Oct 5, 2022 by zhangruiskyline Loading…
24
Back out "Enable shard storage estimator to consider multiple optimizer classes" CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#684 opened Oct 4, 2022 by GD06 Loading…
1
Decouple compute kernel from sharding CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#678 opened Sep 30, 2022 by YLGH Loading…
9
Add a test to ensure feature_processor weights are checkpointed CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#645 opened Sep 19, 2022 by xush6528 Loading…
2
Add helper to apply overlapped optimizer to module CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#638 opened Sep 16, 2022 by lequytra Loading…
1
Fix False -> None for apply_overlapped_optimizer in tests CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#637 opened Sep 16, 2022 by YLGH Loading…
1
Move context creation into input_dist(...) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#610 opened Sep 8, 2022 by xing-liu Loading…
2
Allow DMP to pass prefix fused params to sharded modules CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported fx
#584 opened Aug 20, 2022 by xing-liu Loading…
4
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.