Skip to content

Pull requests: pytorch/torchrec

New pull request New
36 Open 2,916 Closed
36 Open 2,916 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

Add GPU RE for MPZCH module 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
#3185 opened Jul 11, 2025 by kausv Loading…
1
Fix compatibility issue with EmbeddingBag for IEN Publish CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3181 opened Jul 10, 2025 by chelqxz Loading…
5
YAML config support for pipeline benchmarking CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3180 opened Jul 10, 2025 by SSYernar Loading…
2
Dynamic 2D sparse parallel CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3177 opened Jul 9, 2025 by iamzainhuda Loading…
2
Revert D77709565 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
#3173 opened Jul 9, 2025 by jd7-tr Loading…
1
batched_embedding_kernel int32 support behind jk CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3164 opened Jul 6, 2025 by basilwong Loading…
4
Add tensorboard to display training and evaluation metrics CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3163 opened Jul 6, 2025 by lizhouyu Loading…
3
Operator level microbenchmarking CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3154 opened Jul 2, 2025 by SSYernar Loading…
2
Support get/set the whole row of metaheader+weight+optimizer from backend for checkpoint saving/loading CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3153 opened Jul 2, 2025 by bobbyliujb Loading…
8
Support get/set the whole row of metaheader+weight+optimizer from backend for checkpoint saving/loading CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3148 opened Jul 1, 2025 by bobbyliujb Loading…
5
Adding support for MOMENTUM_DIFF and ROWWISE_ADAGRAD optimizer states CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3144 opened Jun 27, 2025 by aliafzal Loading…
1
Adding support for tracking optimizers states in Model Delta Tracker. CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3143 opened Jun 27, 2025 by aliafzal Loading…
1
Add input dist to Perf CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3138 opened Jun 25, 2025 by gregmacnamara Loading…
1
cache offsets / lengths on updates CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3135 opened Jun 24, 2025 by murphymatt Loading…
2
Zero Collision Hash Benchmark Framework CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3127 opened Jun 23, 2025 by lizhouyu Loading…
7
Simplify fqns_to_feature_names in Delta Tracker CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3081 opened Jun 11, 2025 by kausv Loading…
5
Benchamrk framework for torchrec CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3072 opened Jun 9, 2025 by lizhouyu Loading…
6
Create an example for MPZCH CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#3063 opened Jun 9, 2025 by lizhouyu Loading…
3
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
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
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
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
Back out "Fuse "states" level tensors to reduce all gather during metrics compute" CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2893 opened Apr 17, 2025 by iamzainhuda Loading…
1
ProTip! Adding no:label will show everything without a label.