Skip to content

Pull requests: pytorch/torchrec

New pull request New
76 Open 3,059 Closed
76 Open 3,059 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
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Integrating PlanLoader within OSS Planner CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported meta-exported
#3376 opened Sep 14, 2025 by aliafzal Loading…
1
multiplethread pipeline CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3370 opened Sep 11, 2025 by zhaozheng09 Loading…
3
Fix the assignment logic of the variable "self.num_batches" in RandomRecDataset CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3364 opened Sep 9, 2025 by hegongshan Loading…
add test coverage for loading state dict from cpu for EBC 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
#3362 opened Sep 9, 2025 by TroyGarden Loading…
2
fix test CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3361 opened Sep 9, 2025 by jd7-tr Loading…
3
Implement ConfigeratorPlanLoader with robust error handling CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported meta-exported
#3356 opened Sep 5, 2025 by aliafzal Loading…
6
PlanLoader addition into planner CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported meta-exported
#3355 opened Sep 5, 2025 by aliafzal Loading…
8
Create new API for JaggedTensor to dense conversion with optimized performance CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3353 opened Sep 5, 2025 by jd7-tr Loading…
1
Add support for Write 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
#3347 opened Sep 3, 2025 by kausv Loading…
1
Update sharding type assertions to support COLUMN_WISE 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
#3344 opened Sep 3, 2025 by jd7-tr Loading…
1
Allow the ability for uneven row wise sharding based on number of buckets for zch CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3341 opened Sep 2, 2025 by nipung90 Loading…
2
Integrating planner stats db with ConfigeratorStats CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported meta-exported
#3331 opened Aug 29, 2025 by aliafzal Loading…
11
Implement __torch_function__ for KeyedTensor CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3329 opened Aug 28, 2025 by jd7-tr Loading…
5
Add ConfigeratorStats to store sharding plan in config store CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported meta-exported
#3327 opened Aug 28, 2025 by aliafzal Loading…
17
Add a flag to disable fallback in mpzch and return empty rows for cache missed ids. CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3325 opened Aug 27, 2025 by yixin94 Loading…
1
Fix init param test for EBC CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3322 opened Aug 26, 2025 by kausv Loading…
1
Empty shard support CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3319 opened Aug 25, 2025 by jd7-tr Loading…
2
TWA syntax error CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3316 opened Aug 25, 2025 by jeffkbkim Loading…
1
enable KEY_VALUE kernel in ManagedCollisionEmbeddingCollectionSharder CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3307 opened Aug 21, 2025 by iamzainhuda Loading…
1
Fix prefetch train pipeline CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3306 opened Aug 21, 2025 by peterfu0 Loading…
2
Add custom all2all interface CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3297 opened Aug 19, 2025 by peterfu0 Loading…
1
directly pass update_util as int flag without syncing iter CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3293 opened Aug 18, 2025 by Fiery Loading…
1
Integrating resharding API to training pipeline CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3289 opened Aug 15, 2025 by isururanawaka Loading…
1
Add --pre argument to fbgemm installer for OSS unit 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
#3288 opened Aug 15, 2025 by nipung90 Loading…
2
Back out "Update KJT stride calculation logic to be based off of inverse_indices for VBE KJTs." CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3286 opened Aug 15, 2025 by nipung90 Loading…
2
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.