-
Notifications
You must be signed in to change notification settings - Fork 528
Pull requests: pytorch/torchrec
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Add sanity check for stride_per_key_per_rank's dim
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#3124
opened Jun 20, 2025 by
jd7-tr
Loading…
Migrate to correct logger interface (#2996)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#3123
opened Jun 20, 2025 by
TroyGarden
Loading…
Add round-trip test for KeyedJaggedTensor to_dict and from_dict
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3122
opened Jun 20, 2025 by
vaishnavirao24
Loading…
Add usage example to KeyedJaggedTensor docstring
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#3121
opened Jun 20, 2025 by
vaishnavirao24
Loading…
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
#3119
opened Jun 20, 2025 by
jd7-tr
Loading…
Added model wrappers for DeepFM and DLRM
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#3115
opened Jun 18, 2025 by
SSYernar
Loading…
kvzch use new operator in model 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
#3108
opened Jun 17, 2025 by
chenyuzhcy
Loading…
Added Optimizer configuration that supports optimizer type, learning rate, momentum, and weight decay configurations.
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#3107
opened Jun 17, 2025 by
SSYernar
Loading…
Support ebc for kv 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
#3106
opened Jun 17, 2025 by
EddyLXJ
Loading…
Added model config for supporting SparseNN, TowerSparseNN, TowerCollectionSparseNN models.
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#3104
opened Jun 17, 2025 by
SSYernar
Loading…
Added Optimizer configuration that supports optimizer type, learning rate, momentum, and weight decay configurations.
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#3094
opened Jun 13, 2025 by
SSYernar
Loading…
Change job name while uploading sharding plan to Manifold
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#3092
opened Jun 13, 2025 by
aporialiao
Loading…
OSS TorchRec Internal MPZCH modules (#3017)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#3089
opened Jun 12, 2025 by
lizhouyu
Loading…
add support for List[torch.Tensor]
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#3088
opened Jun 12, 2025 by
iamzainhuda
Loading…
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…
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…
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…
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…
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…
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…
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…
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…
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…
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…
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…
Previous Next
ProTip!
Updated in the last three days: updated:>2025-06-17.