All pull requests

Pull requests

• Propagate transfer parallelism into tree-broadcast hop plans.

  #1085·copybara-service[bot] opened Sep 28, 2026Bot·

  ·

• [XLA] Bump OpenXLA to f60be94c9c5d

  pull ready
  #1084·jparkerh opened Sep 28, 2026Contributor·

  ·

• Prototype: opt-in flags to drop libbpf, libacl and zeromq from GGC builds

  #1083·copybara-service[bot] opened Sep 28, 2026Bot·

  ·
  1

• Fixed weights sync tree broadcast checksum mismatch across multi-stage pipelines.

  #1082·copybara-service[bot] opened Sep 28, 2026Bot·

  ·

• Add CPU repro test for tree-broadcast seed/relay completion defects.

  #1081·copybara-service[bot] opened Sep 28, 2026Bot·

  ·

• Add a TPU7x test that reproduces compute / DMA overlal raw DMA kernel stalls.

  #1080·copybara-service[bot] opened Sep 27, 2026Bot·

  ·

• Pace tpu_raiden torch raw DMA so kernels don't wait for whole copies.

  #1079·copybara-service[bot] opened Sep 27, 2026Bot·

  ·

• Implement bandwidth-matched All-Source Binomial Tree broadcast and C++ round-grouped destination logging.

  #1078·copybara-service[bot] opened Sep 27, 2026Bot·

  ·

• Added TPU Raiden weight synchronization roofline analysis suite, CLI, interactive notebook, and agent skill.

  #1077·copybara-service[bot] opened Sep 27, 2026Bot·

  ·

• Bump tpu_raiden JAX build to 0.11.2 and support runtime JAX >= 0.11.0

  #1076·copybara-service[bot] opened Sep 27, 2026Bot·

  ·

• Decouple address-free resharding plan from transfer plan and use 1-to-1 whole-block sampler relays.

  #1072·copybara-service[bot] opened Sep 27, 2026Bot·

  ·

• Add PyTorch unit test suites (tpu_sync/frameworks/torch:all and tpu_sync/api/torch:all) to TAP blueprint test_patterns.

  #1071·copybara-service[bot] opened Sep 27, 2026Bot·

  ·
  1

• Instantiate tpu_raiden::jax::WeightSynchronizer in JAX weight synchronizer FFI modules.

  #1070·copybara-service[bot] opened Sep 27, 2026Bot·

  ·

• Add bounded host ring-buffer pools and DP sub-sharding with post-H2D ICI all-gather.

  #1069·copybara-service[bot] opened Sep 27, 2026Bot·

  ·

• Add an opt-in cache of shard files confirmed on storage to

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

  ·

• Implement full-duplex chunked pipelining across the All-Source Binomial Tree.

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

  ·

• 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