9 Pull requests opened by 6 people

• add virtual size support for shardedTensor

  #2915 opened Apr 24, 2025

• support zero collision tables in ssd operator

  #2919 opened Apr 28, 2025

• torchrec support on kvzch emb lookup module

  #2922 opened Apr 28, 2025

• Update the _maybe_compute_stride_kjt logic to calculate stride based off of inverse_indices for VBE KJTs.

  #2925 opened Apr 29, 2025

• Allow a context manager to be called around apply_jit

  #2927 opened Apr 30, 2025

• Add enable_raw_embedding_streaming from TBE config to EmbeddingKVDB

  #2928 opened Apr 30, 2025

• provide basic integration of trec 2d emb to training pipeline

  #2929 opened Apr 30, 2025

• Add logic to stream weights in EmbeddingKVDB

  #2930 opened Apr 30, 2025

• Extend FSDP1 global clipping support for optimizers other than Shampoo

  #2931 opened May 1, 2025

1 Issue opened by 1 person

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

  #2923 opened Apr 29, 2025

1 Unresolved conversation

Sometimes conversations happen on old items that aren’t yet closed. Here is a list of all the Issues and Pull Requests with unresolved conversations.

• Test include stride_per_key to KJT's flatten and unflatten

  #2903 commented on May 1, 2025 • 0 new comments