-
Notifications
You must be signed in to change notification settings - Fork 642
Pull requests: meta-pytorch/torchrec
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Override optimizer to EXACT_ROWWISE_ADAGRAD for cache-mode KVZCH TBEs
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4213
opened May 3, 2026 by
XialuLi
Loading…
feat: add per-TBE tbe_composition event
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4212
opened May 1, 2026 by
micrain
Loading…
construct_jagged_tensors index_select change (#4211)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4211
opened May 1, 2026 by
jian-he
Loading…
Fix KTRegroupAsDict _out_lengths None guard and FX/TorchScript compat
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4210
opened May 1, 2026 by
sacdroid
Loading…
Add sharded relay tests and benchmarks (#4209)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4209
opened May 1, 2026 by
srinathb-meta
Loading…
Add sharded relay distributed utilities (#4207)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4207
opened May 1, 2026 by
srinathb-meta
Loading…
Emit Scuba event on collective tag mismatch
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4206
opened May 1, 2026 by
kaanbaloglu
Contributor
Loading…
Enable SDD Lite and RecMetric Benchmark In Nightly Jobs (#4204)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4204
opened Apr 30, 2026 by
siqihuang
Loading…
Guard _safe_tolist JK check with is_torchdynamo_compiling (#4201)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4201
opened Apr 30, 2026 by
jeffkbkim
Contributor
Loading…
Fix UVM cache cross-stream race in PrefetchTrainPipelineSparseDist (#4191)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4191
opened Apr 29, 2026 by
kaanbaloglu
Contributor
Loading…
Adding logger to detect Topology creation via TopologyFactory and Issue Warnings (#4189)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4189
opened Apr 28, 2026 by
isururanawaka
Contributor
Loading…
Add FixedAbsoluteStorageReservation for absolute GB/bytes HBM reservation (#4182)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4182
opened Apr 27, 2026 by
aporialiao
Contributor
Loading…
Add SHAMPOO_V3_MRS to OptimType enum
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4174
opened Apr 27, 2026 by
Shagun-G
Loading…
KJT: centralize tensor enumeration in _all_tensors() helper (#4172)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4172
opened Apr 27, 2026 by
kaanbaloglu
Contributor
Loading…
KJT record_stream and clear_storage: cover _inverse_indices and _jt_dict (#4171)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4171
opened Apr 27, 2026 by
kaanbaloglu
Contributor
Loading…
Optimizer config and MRS wrapper for shampoo v3 in MVAI
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4168
opened Apr 24, 2026 by
Shagun-G
Loading…
Fix ThroughputMetric batch_size in train-eval phase transitions
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4167
opened Apr 23, 2026 by
alyssawangqq
Loading…
Add sync_dataloader_exhaustion to TrainPipelineSparseDist (#4158)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4158
opened Apr 22, 2026 by
kaanbaloglu
Contributor
Loading…
Add FP8 rowwise padding to quantized AllToAll pooled embeddings (#4153)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4153
opened Apr 22, 2026 by
RohanVardhan
Loading…
Fix optimizer state sizing to use float32 instead of weight dtype
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4142
opened Apr 21, 2026 by
jeffkbkim
Contributor
Loading…
Disable shape_padding in TorchRec TrainPipelineSparseDist family + TrainPipelinePT2 to prevent cross-PG deadlock on AMD (#4128)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4128
opened Apr 16, 2026 by
kaanbaloglu
Contributor
Loading…
Scale batch_outputs by min(1, input_length)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4125
opened Apr 16, 2026 by
gregmacnamara
Contributor
Loading…
Add fused KJT data A2A across EBC modules (#4120)
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4120
opened Apr 16, 2026 by
jeffkbkim
Contributor
Loading…
Add Error Handling In total_average of Pytorch Profiler In Perf Benchmark
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4105
opened Apr 14, 2026 by
siqihuang
Loading…
Conditionally import non-OSS packages
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
meta-exported
#4092
opened Apr 11, 2026 by
Frederick-Zhu
Loading…
Previous Next
ProTip!
Exclude everything labeled
bug with -label:bug.