All pull requests

Pull requests

• Upgrade packages to Pyrefly R1543.1 (landing group 82)

  CLA Signedmeta-exported
  #4993·meta-codesync[bot] opened Nov 1, 2026Bot·

  ·
  1

• Deduplicate mutating LRU lookup inputs (#4991)

  CLA Signedmeta-exported
  #4991·jiachen247 opened Oct 31, 2026·

  ·
  1

• Handle empty shard owners in TPU VBE all-to-all

  CLA Signedmeta-exported
  #4990·Ali-Tehrani opened Oct 31, 2026Contributor·

  ·
  1

• Restore metadata-aware KV checkpoints through PMT

  CLA Signedmeta-exported
  #4989·q10 opened Oct 30, 2026Contributor·

  ·
  1

• Fix forward-only DMP initialization for FSDP2 models

  CLA Signedmeta-exported
  #4988·xywang9334 opened Oct 30, 2026·

  ·
  1

• [TorchRec][preserve-forward-order-during-backward] Preserve sparse forward order during backward

  CLA Signed
  #4987·fuzailm8 opened Oct 30, 2026·

  ·

• Row writes into the DRAM KV enrichment cache

  CLA Signedmeta-exported
  #4986·txWang opened Oct 30, 2026·

  ·

• [TorchRec][sync-only-used-rows-2d] Expose used-row module sync in the train pipeline

  CLA Signed
  #4984·fuzailm8 opened Oct 30, 2026·

  ·

• [TorchRec][sync-only-used-rows-2d] Synchronize packed used embedding rows

  CLA Signed
  #4983·fuzailm8 opened Oct 30, 2026·

  ·

• [TorchRec][sync-only-used-rows-2d] Track replica-union used embedding rows

  CLA Signed
  #4982·fuzailm8 opened Oct 30, 2026·

  ·

• [TorchRec][preserve-forward-order-during-backward] Add sparse backward priority runtime

  CLA Signed
  #4981·fuzailm8 opened Oct 30, 2026·

  ·

• [TorchRec][per-module-sync] Wait for 2D module state at its consumers

  CLA Signed
  #4980·fuzailm8 opened Oct 30, 2026·

  ·

• [TorchRec][per-module-sync] Launch 2D module sync from fused updates

  CLA Signed
  #4979·fuzailm8 opened Oct 30, 2026·

  ·

• [TorchRec][per-module-sync] Add module-granular DMP replica synchronization

  CLA Signed
  #4978·fuzailm8 opened Oct 30, 2026·

  ·

• [TorchRec][per-module-sync] Discover planned module sync ownership

  CLA Signed
  #4977·fuzailm8 opened Oct 30, 2026·

  ·

• Sync torch_tpu SparseCore embedding updates

  CLA Signed
  #4976·songbai-blip opened Oct 29, 2026·

  ·
  2

• Unwrap stale pipelined forwards in _rewrite_model.

  CLA Signedmeta-exported
  #4975·murray opened Oct 29, 2026Contributor·

  ·
  1

• Account for physical fused RowWiseAdagrad states in LP planning

  CLA Signedmeta-exported
  #4970·zurl opened Oct 29, 2026·

  ·
  1

• Plumb res_hbm_drain_interval_per_table to RESParams (#4963)

  CLA Signedmeta-exported
  #4963·FriedCosey opened Oct 28, 2026·

  ·
  1

• Validate UVM cache padding

  CLA Signedmeta-exported
  #4962·bharathvemula opened Oct 28, 2026·

  ·
  1

• Cover rank-aware UVM padding

  CLA Signedmeta-exported
  #4961·bharathvemula opened Oct 28, 2026·

  ·
  1

• Account for UVM cache padding (#4960)

  CLA Signedmeta-exported
  #4960·bharathvemula opened Oct 28, 2026·

  ·
  1

• Respect cache precision in UVM HBM estimates (#4959)

  CLA Signedmeta-exported
  #4959·bharathvemula opened Oct 28, 2026·

  ·
  1

• Remove deprecated size-oblivious APIs

  CLA Signedmeta-exported
  #4950·laithsakka opened Oct 26, 2026Contributor·

  ·
  1

• Fix multi-group all-to-all and input dist costs; document multi-group vs GRID_SHARD and ROW_WISE

  CLA Signedmeta-exported
  #4947·dmdok opened Oct 26, 2026Contributor·

  ·
  1