All pull requests

Pull requests

• [tpu_sync] Drop the Peregrine mock in the tests so tpu_sync no longer implements Peregrine's transport interface.

  #1267·copybara-service[bot] opened Oct 16, 2026Bot·

  ·

• Replaced host_subgrid heuristics with explicit shard placement.

  #1266·copybara-service[bot] opened Oct 16, 2026Bot·

  ·

• [peregrine] deprecate src/api/socket_util.h

  #1265·copybara-service[bot] opened Oct 15, 2026Bot·

  ·
  1

• [tpu_raiden] Update peregrine dependency to v0.1.6

  #1264·copybara-service[bot] opened Oct 15, 2026Bot·

  ·
  1

• KVCacheStore runs in the vLLM scheduler process, where LOCAL_RANK is unset.

  #1262·copybara-service[bot] opened Oct 15, 2026Bot·

  ·

• ci: add jax_version workflow_dispatch input to wheel workflows

  #1261·linamy85 opened Oct 15, 2026Collaborator·

  ·
  1

• Fail the pool-reshard receive session on a failed incoming push so its staging lease and plan claim are released immediately instead of at the session deadline.

  #1259·copybara-service[bot] opened Oct 14, 2026Bot·

  ·
  1

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

  #1257·copybara-service[bot] opened Oct 14, 2026Bot·

  ·

• 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·

  ·

• Add fault injection hooks for tpu_sync DMA dispatch paths.

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

  ·
  1

• Await already-started copies before reporting partial-start errors.

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

  ·
  1

• Implement Async methods in TdsKVBackend, and wire it in KV-manager briefly over libtdsul.

  #1222·copybara-service[bot] opened Oct 10, 2026Bot·

  ·
  1

• Keep the PJRT client alive until torch raw buffer aliases are destroyed

  #1199·norx1991 opened Oct 6, 2026·

  ·
  1

• Let a pool reshard transfer name any subset of source ranks

  #1180·jinzhenfan opened Oct 5, 2026·

  ·

• Added the controller v3 Python controller and remote facade.

  #1179·copybara-service[bot] opened Oct 5, 2026Bot·

  ·

• Exposed controller v3 to Python with typed request and slice helpers.

  #1178·copybara-service[bot] opened Oct 5, 2026Bot·

  ·

• Added the controller v3 C++ controller and its pull server.

  #1177·copybara-service[bot] opened Oct 5, 2026Bot·

  ·

• Added controller v3 pull scheduling and the dynamic pull engine.

  #1176·copybara-service[bot] opened Oct 5, 2026Bot·

  ·

• Added the controller v3 logical reshard planner.

  #1175·copybara-service[bot] opened Oct 5, 2026Bot·

  ·

• Add COMMAND_START_SWARM_SESSION to start a swarm session on a listener

  #1170·copybara-service[bot] opened Oct 5, 2026Bot·

  ·

• Add swarm bundle-pull functional test with a trainer-less seed sampler

  #1169·copybara-service[bot] opened Oct 5, 2026Bot·

  ·