Search results

System OOM Error in Distributed Checkpoint Writer When using fused_uvm_caching Kernel

Status: Open.

#3399 In meta-pytorch/torchrec;

· tiankongdeguiji opened on Sep 29, 2025

confused with torchrec/distributed/sharding/dynamic_sharding.py

Status: Open.

#3395 In meta-pytorch/torchrec;

· haolujun opened on Sep 26, 2025

Allow missing keys in EmbeddingBagCollection forward

Status: Open.

#3305 In meta-pytorch/torchrec;

· rayhuang90 opened on Aug 21, 2025

invoke_on_rank_and_broadcast_result stuck

Status: Open.

#3278 In meta-pytorch/torchrec;

· 121547456 opened on Aug 13, 2025

DistributedModelParallel sets "requires_grad" to False for EmbeddingCollection with sharding_types=["row_wise"], compute_kernels=["fused"],

Status: Open.

#3271 In meta-pytorch/torchrec;

· farapart opened on Aug 11, 2025

Working with uneven number of batches per rank/worker in inference

Status: Open.

#3132 In meta-pytorch/torchrec;

· bkumar4-sc opened on Jun 24, 2025

Which lightning strategy to use with torchrec optimizers?

Status: Open.

#3114 In meta-pytorch/torchrec;

· JacobHelwig opened on Jun 18, 2025

[Bug]: ManagedCollisionEmbeddingCollection returns all-zero embeddings after applying apply_optimizer_in_backward with RowWiseAdagrad

Status: Open.

#3030 In meta-pytorch/torchrec;

· rayhuang90 opened on Jun 3, 2025

ManagedCollisionEmbeddingBagCollectionSharder cannot support CPU

Status: Open.

#2958 In meta-pytorch/torchrec;

· YuandongXu-hotstar opened on May 8, 2025

CUDA error: an Illegal memory access was encountered

Status: Open.

#2957 In meta-pytorch/torchrec;

· ali-fani-sd opened on May 7, 2025

Can NVIDIA-Merlin/HierarchicalKV integrated to torchrec as EmbeddingBag Extension?

Status: Open.

#2923 In meta-pytorch/torchrec;

· hudengjunai opened on Apr 29, 2025

how to freeze embedding

Status: Open.

#2908 In meta-pytorch/torchrec;

· threestone965 opened on Apr 23, 2025