Skip to content

Pull requests: pytorch/torchrec

New pull request New
Clear current search query, filters, and sorts
312 Open 2,324 Closed
312 Open 2,324 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

Sequence embedding TWRW 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
#166 opened Mar 21, 2022 by bigning Loading…
2
Benchmark batching funcs CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#187 opened Mar 31, 2022 by s4ayub Loading…
1
Filter out batch_fused from available kernels if fused_params is not explicitly set and contains optimizer CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#298 opened May 5, 2022 by YLGH Loading…
1
Update Lazy Extension so in sync with Pytorch CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#339 opened May 16, 2022 by dstaay-fb Loading…
1
Repro the ghost processes for TorchAsyncITer CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#521 opened Jul 12, 2022 by RenfeiChen-FB Loading…
1
Support sparse gradient for embedding bag collection CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#522 opened Jul 12, 2022 by mmenz Loading…
1
add JaggedTensorMeta CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#532 opened Jul 21, 2022 by yinbinm Loading…
1
add get_fused_optimizers CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#558 opened Aug 1, 2022 by YLGH Loading…
1
Modify torchrec code to use reduce_scatter_v and all_gather_v CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#568 opened Aug 5, 2022 by YazhiGao Loading…
1
Disable sub-group pg on RDMA hosts CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#573 opened Aug 11, 2022 by xw285cornell Loading…
2
fix tower 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
#576 opened Aug 15, 2022 by bigning 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
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
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
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
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
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
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
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
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
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
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
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
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…
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
ProTip! What’s not been updated in a month: updated:<2025-02-24.