-
Notifications
You must be signed in to change notification settings - Fork 514
Pull requests: pytorch/torchrec
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Harden Tower QPS Fused Mode
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
<
10000
/div>
#2992
opened May 21, 2025 by
HugeEngine
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…
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…
: pe tasks
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2988
opened May 21, 2025 by
dragonxlwang
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…
Refactoring ITEP / PTP Pruning Scuba Logger [1/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
#2986
opened May 21, 2025 by
doIIarplus
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…
split train_pipeline.utils - tracing
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2982
opened May 19, 2025 by
TroyGarden
Loading…
support get state dict and apply state dict
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2976
opened May 17, 2025 by
emlin
Loading…
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…
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…
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…
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…
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…
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…
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…
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…
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…
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…
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…
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…
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…
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…
Previous Next
ProTip!
Type g i on any issue or pull request to go back to the issue listing page.