All pull requests

Pull requests

• Add offline symbolic and parallel weight-sync plan computation.

  #1066·copybara-service[bot] opened Sep 26, 2026Bot·

  ·

• Reshard KV pools between pipeline-parallel engines

  #1064·copybara-service[bot] opened Sep 26, 2026Bot·

  ·

• - Replace the single-slot ShardScratchpad in WeightSynchronizerBase with a 2-slot (kNumScratchpadSlots = 2) double-buffered scratchpad per shard (ScratchpadSlot slots[2]), allowing CPU tiling/detiling on one slot to overlap with in-flight PCIe H2D/D2H DMA (CopyRawHostToDevice / CopyRawDeviceToHost) on the other slot without serializing on in_flight_future.Await().

  #1063·copybara-service[bot] opened Sep 26, 2026Bot·

  ·

• Reshard KV pools between pipeline-parallel engines

  #1062·yinlin09 opened Sep 25, 2026Collaborator·

  ·

• [tpu_sync] Support multi-backend Load() with secondary storage suffix in KVCacheStore.

  #1061·copybara-service[bot] opened Sep 25, 2026Bot·

  ·

• [tpu_sync] Replace copied Peregrine code with peregrine v0.1.1 dependency

  #1054·copybara-service[bot] opened Sep 25, 2026Bot·

  ·
  1

• [tpu_raiden] Refactor block transport to Peregrine-like post/poll API (13/N)

  #1050·copybara-service[bot] opened Sep 25, 2026Bot·

  ·

• Allow configuring fault injection rules via a plain text file. When the file is absent - no rules are installed.

  #1045·copybara-service[bot] opened Sep 25, 2026Bot·

  ·
  1

• [TPU Raiden] Fix [Failure/ReceiveDisconnect]: propagate broken inbound streams to fail active receive sessions and release staging slots immediately.

  #1041·copybara-service[bot] opened Sep 24, 2026Bot·

  ·
  1

• Remove worker_rpc_client parameter and encapsulate WorkerRpcClient in JobEntity

  #1027·copybara-service[bot] opened Sep 23, 2026Bot·

  ·

• Park pulls awaiting registration instead of holding a server thread

  pull ready
  #1026·jndevanshu opened Sep 23, 2026Collaborator·

  ·
  

• Add GrpcTransportAdapter with PushBlocks support

  #1025·copybara-service[bot] opened Sep 23, 2026Bot·

  ·

• Optimize tree broadcast hop pre-computation, layer-scoped D2H/H2D, and relay coalescing

  #1024·copybara-service[bot] opened Sep 23, 2026Bot·

  ·

• Settle zero-block/zero-layer receives and ready sessions in WaitForPendingWork

  #995·copybara-service[bot] opened Sep 21, 2026Bot·

  ·

• [Raiden/FaultTolerance] Support chunk header version compatibility validation and version-dependent parsing

  #993·copybara-service[bot] opened Sep 21, 2026Bot·

  ·
  1

• [gmon] Add Monarch retention rules for P2P transfer latency and PCIe byte metrics.

  #959·copybara-service[bot] opened Sep 18, 2026Bot·

  ·

• Declare and instrument PCIe H2D and D2H byte counter and per-device label telemetry metrics:

  #950·copybara-service[bot] opened Sep 18, 2026Bot·

  ·

• Add TdsKVBackend storage driver and wire TDS backend into KVCacheManagerBase and KVCacheStore.

  #948·copybara-service[bot] opened Sep 18, 2026Bot·

  ·
  1

• Internal build configuration cleanup for torch_tpu src/ layout

  #938·copybara-service[bot] opened Sep 17, 2026Bot·

  ·

• Add PyTorch support to TPU Raiden H2D/D2H presubmit performance gate for GitHub Actions.

  #925·copybara-service[bot] opened Sep 14, 2026Bot·

  ·
  1

• ci: build the release matrix from the frameworks input

  #917·yinlin09 opened Sep 13, 2026Collaborator·

  ·

• OptionA experimenting

  #898·wenjung2007 opened Sep 10, 2026·

  ·
  2

• Implement exponential backoff circuit breaker for failing producer peers in tpu_sync.

  #887·copybara-service[bot] opened Sep 8, 2026Bot·

  ·
  1

• Chunk vectored socket I/O at IOV_MAX instead of rejecting large iovec spans

  #885·majunze2001 opened Sep 8, 2026·

  ·
  1

• Implement gRPC-based transfer control plane for KVCacheManagerWithTransfer.

  #881·copybara-service[bot] opened Sep 8, 2026Bot·

  ·