Skip to content

Pull requests: meta-pytorch/torchrec

New pull request New
173 Open 4,023 Closed
173 Open 4,023 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

Preserve unquantized embedding tables through sharding CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4463 opened Jul 24, 2026 by shawnxu26 Loading…
1
Fix transfer_tensors_to_cpu crash on mixed CPU/CUDA dicts CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4461 opened Jul 24, 2026 by jeffkbkim Contributor Loading…
1
Add skip_batch_waiting option to skip _wait_for_batch() in customized-order sparse-dist train pipeline CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4460 opened Jul 23, 2026 by Fiery Loading…
1
Add preserving stride per key per rank to KJT.from_jt CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4459 opened Jul 23, 2026 by Ali-Tehrani Contributor Loading…
1
Support VBE in GRID_SHARD output-dist CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4456 opened Jul 22, 2026 by dmdok Loading…
1
(4/N)[MPZCH] Fresh-wins read path for the Main/Fresh region split CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4454 opened Jul 21, 2026 by qyxxxigelu Loading…
1
Expose ProcessGroupNCCL internal comm stream to Python CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4453 opened Jul 21, 2026 by TroyGarden Contributor Loading…
1
Route APF sharding-plan through create_sharding_plan (JK-gated) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4451 opened Jul 21, 2026 by isururanawaka Contributor Loading…
1
Enable supports_test_execution_caching for test_cpu_offloaded_metric_module CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4448 opened Jul 20, 2026 by johnny90 Contributor Loading…
1
memory_stashing: avoid double-buffering on stash/restore overlap CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4447 opened Jul 20, 2026 by doIIarplus Loading…
1
Fix int32 overflow in variable-batch KJT all-to-all _get_recat CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4443 opened Jul 17, 2026 by ycui1984 Loading…
1
Add async_inplace_copy: offload in-place H2D copy dispatch to a background thread CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4441 opened Jul 17, 2026 by aporialiao Contributor Loading…
1
(3/N)[MPZCH] Add write_to_fresh_region to target Main or Fresh on insert CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4440 opened Jul 17, 2026 by qyxxxigelu Loading…
1
(2/N)[MPZCH] Require disable_fallback and no opt-in for a Region Split CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4438 opened Jul 17, 2026 by qyxxxigelu Loading…
1
Add Main/Fresh region split (percent_fresh_region) (#4436) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4436 opened Jul 17, 2026 by qyxxxigelu Loading…
1
Fix _collective_tag_from being too expensive on wide RW-sharded models (#4418) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4418 opened Jul 10, 2026 by kaanbaloglu Contributor Loading…
1
Reland D110435026 with non-DI int4 (quint4x2) merge-sample lowering fix CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4417 opened Jul 9, 2026 by zhaozhul Contributor Loading…
1
Allocate pairwise-distinct train pipeline streams CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4396 opened Jun 30, 2026 by jeffdaily Loading…
1
Support value gradient clipping for sharded params (FSDP/HSDP) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4388 opened Jun 25, 2026 by wz337 Contributor Loading…
1
Make stash forward hook compile-safe via forward-patching CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4383 opened Jun 25, 2026 by aporialiao Contributor Loading…
1
Fix MP-ZCH benchmark FP16_table buckets for world_size=8 CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4369 opened Jun 22, 2026 by spmex Loading…
1
defensive snapshot of model_out via torch._foreach_clone CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported
#4341 opened Jun 11, 2026 by jeffkbkim Contributor Loading…
1
Load ir_metadata_tensor from bytearray CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4336 opened Jun 10, 2026 by pganssle-google Loading…
Fix broken fbcode//torchrec/distributed/tests:test_dynamic_sharding - test_dynamic_sharding_ebc_tw ( 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
#4323 opened Jun 4, 2026 by hammad45 Loading…
1
fix: add ALBT to OSS TorchrecComponent enum (#4318) 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
#4318 opened Jun 4, 2026 by metahongw Contributor Loading…
1
ProTip! Follow long discussions with comments:>50.