Skip to content

Pull requests: pytorch/torchrec

New pull request New
341 Open 2,446 Closed
341 Open 2,446 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

Fix shape issue with remapped indices in sharded ManagedCollisionEmbeddingCollection CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3010 opened May 28, 2025 by arthisuresh Loading…
1
Add hashing for ParameterConstraints CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3009 opened May 28, 2025 by micrain Loading…
2
add emb cast in KTRegroupAsDict 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
#3008 opened May 28, 2025 by TroyGarden Loading…
5
Handle Virtual Table sizing in 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
#3004 opened May 28, 2025 by kausv Loading…
5
Refactoring ITEP / PTP Pruning Scuba Logger [3/N] CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3002 opened May 27, 2025 by doIIarplus Loading…
10
Migrate to correct logger interface CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2996 opened May 23, 2025 by emmanuel-ferdman Loading…
Enable fx tracing for maybe_td_to_kjt CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2991 opened May 21, 2025 by jeetkanjani7 Loading…
1
Fix undeclared type in colab for sharding examples CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2989 opened May 21, 2025 by lizhouyu Loading…
1
Support RecMetrics new options in Pyper config CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2987 opened May 21, 2025 by HugeEngine Loading…
2
mark metricmodule methods as experimental CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2983 opened May 19, 2025 by iamzainhuda Loading…
2
Revert D74293458 ci-no-td CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2974 opened May 16, 2025 by BoyuanFeng Loading…
1
Fix torchrec type 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
#2973 opened May 16, 2025 by aporialiao Loading…
1
Verify Pyre version CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2972 opened May 15, 2025 by aporialiao Loading…
1
Fix Pyre CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2970 opened May 15, 2025 by aporialiao Loading…
2
Support DI sharding for FPE_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
#2968 opened May 13, 2025 by hannaxu Loading…
2
Swap CPU TBE lookup kernel with SSD TBE lookup kernel for tables being offloaded to SSD CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2965 opened May 13, 2025 by faran928 Loading…
1
convert stride_per_key_per_rank to tensor inside KJT CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2959 opened May 8, 2025 by TroyGarden Loading…
9
Add missing fields to KJT's PyTree flatten/unflatten logic for VBE KJT CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2952 opened May 7, 2025 by jd7-tr Loading…
1
Create a new tensor arg for stride_per_key_per_rank to facilitate torch.export CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2950 opened May 6, 2025 by jd7-tr Loading…
2
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
#2949 opened May 6, 2025 by jd7-tr Loading…
3
Add missing Pyre mode headers] [shard:2/N] CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2947 opened May 6, 2025 by facebook-github-bot Loading…
1
DistributedModelParallel resharding 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
#2945 opened May 6, 2025 by aporialiao Loading…
12
Extend FSDP1 global clipping support for optimizers other than Shampoo CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2931 opened May 1, 2025 by wz337 Loading…
1
Allow a context manager to be called around apply_jit CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2927 opened Apr 30, 2025 by kddnewton Loading…
4
Update the _maybe_compute_stride_kjt logic to calculate stride 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
#2925 opened Apr 29, 2025 by jd7-tr Loading…
3
ProTip! Follow long discussions with comments:>50.