All pull requests

Pull requests

• Combine h2d_d2h_gating_jax and h2d_d2h_gating_torch into a single TPU benchmark job (--framework=all) in benchmark_registry.pbtxt, ci/run_in_torch_env.py, and copy.bara.sky.

  #1484·copybara-service[bot] opened Nov 4, 2026Bot·

  ·

• Use pre-baked ml-build:gcc12-headers-ibverbs-bazel container image in CI workflows and benchmark registry, and skip runtime apt-get and Bazel bootstrap downloads when pre-installed.

  #1483·copybara-service[bot] opened Nov 4, 2026Bot·

  ·

• Fold the E2E JAX build targets into the Phase 1 bazel test invocation in ci/run_unit_tests.sh and skip redundant pip install -r requirements.txt.

  #1481·copybara-service[bot] opened Nov 4, 2026Bot·

  ·

• Add --only-extensions and --skip-pip flags to build.sh and pass them in ci/run_in_torch_env.py to skip unused C++ server targets and redundant requirements.txt installation during PyTorch CI benchmark builds.

  #1480·copybara-service[bot] opened Nov 4, 2026Bot·

  ·

• Run h2h_cpp_gate on the CPU runner pool (linux-x86-n2-32) with environment ID cpu-n2-32 in benchmark_registry.pbtxt.

  #1479·copybara-service[bot] opened Nov 4, 2026Bot·

  ·

• [tpu_raiden] Update peregrine dependency to v0.1.15

  #1477·copybara-service[bot] opened Nov 4, 2026Bot·

  ·

• Add intra-worker NUMA sub-manager aggregation to the Lean formal verification.

  #1476·copybara-service[bot] opened Nov 3, 2026Bot·

  ·

• Scope receiver push-progress assumptions and replay transport, send-slot, and routing unit tests in the Lean formal verification.

  #1475·copybara-service[bot] opened Nov 3, 2026Bot·

  ·

• Automated Code Change

  #1471·copybara-service[bot] opened Nov 3, 2026Bot·

  ·

• Simplify the Lean formal verification models to decouple candidate fixes from shipping behavior.

  #1469·copybara-service[bot] opened Nov 3, 2026Bot·

  ·

• Implement true TDS-tunnel DMA-BUF P2P direct storage path in Raiden.

  #1463·copybara-service[bot] opened Nov 3, 2026Bot·

  ·
  1

• tpu_raiden: Add 4-rank MPMD e2e test for out-of-band secondary storage eviction and corruption

  #1461·copybara-service[bot] opened Nov 2, 2026Bot·

  ·
  1

• Add tpu_raiden_storage_filesystem_bytes gauge for POSIX secondary storage capacity.

  #1458·copybara-service[bot] opened Nov 2, 2026Bot·

  ·

• Serve DMA mapping from the Mooncake TPU buffer provider.

  #1445·copybara-service[bot] opened Nov 2, 2026Bot·

  ·
  3

• Internal

  #1374·copybara-service[bot] opened Oct 28, 2026Bot·

  ·

• In KVCacheStore::Create, RecoverFromLocalManifest runs before store_monitor_ is constructed, so its RequestRepublish call was a no-op when recovering from a warm shared-memory metadata table (RAIDEN_SHM_KEY). Record the number of blocks recovered during Create and invoke store_monitor_->RequestRepublish() once the monitor starts so recovered host-resident blocks are advertised to the global registry on startup.

  #1310·copybara-service[bot] opened Oct 22, 2026Bot·

  ·
  1

• Add PyTorch KVCacheManager functional offload/reload tests

  #1289·copybara-service[bot] opened Oct 19, 2026Bot·

  ·
  1

• Add a TPU buffer provider for Mooncake TENT.

  #1279·copybara-service[bot] opened Oct 19, 2026Bot·

  ·
  1

• Bypass DRAM cache residency in KVCacheStore when TDS storage caching is active.

  #1278·copybara-service[bot] opened Oct 19, 2026Bot·

  ·
  1

• ci: add jax_version workflow_dispatch input to wheel workflows

  #1261·linamy85 opened Oct 15, 2026Collaborator·

  ·
  1

• Update install instructions to the tpu-sync-{jax,torch} PyPI packages.

  #1245·copybara-service[bot] opened Oct 13, 2026Bot·

  ·

• Reject an invalid read registration with an error instead of 0

  #1242·yinlin09 opened Oct 13, 2026Collaborator·

  ·

• Release idle data connections per transfer (TPU_RAIDEN_DATA_CONN_REUSE) and decouple streams per push from the rank count (TPU_RAIDEN_TRANSFER_STREAMS)

  pull ready
  #1241·sixiang-google opened Oct 13, 2026Collaborator·

  ·
  1

• Report why a transfer failed in poll_stats_with_details

  #1239·yinlin09 opened Oct 13, 2026Collaborator·

  ·

• Introduce StagingArena base class with LayerArena, SlotArena and BlockArena

  #1232·copybara-service[bot] opened Oct 12, 2026Bot·

  ·