Skip to content

Pull requests: pytorch/torchrec

New pull request New
325 Open 2,504 Closed
325 Open 2,504 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 optimizer check edge case CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3058 opened Jun 7, 2025 by aporialiao Loading…
2
set DDR size in planner to 0 CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3052 opened Jun 6, 2025 by kausv Loading…
1
Update column_wise to allow uneven 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
#3050 opened Jun 6, 2025 by jd7-tr Loading…
2
Add hashing for Topology CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3045 opened Jun 5, 2025 by micrain Loading…
2
Add missing Pyre mode headers] CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3044 opened Jun 5, 2025 by facebook-github-bot Loading…
1
During DLRM examples to create prediction using TorchRec library. This shows basic example of how to use TorchRec library quickly locally. CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3043 opened Jun 5, 2025 by yupadhyay Loading…
1
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…
6
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…
12
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
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
Move util function for creating VBE KJT to the VbModelInput class CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2906 opened Apr 22, 2025 by jd7-tr Loading…
1
ProTip! Follow long discussions with comments:>50.