All pull requests

Pull requests

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

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

  ·

• Added pull-lease scheduling and the dynamic pull engine for controller v3.

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

  ·

• Added the controller v3 logical reshard planner.

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

  ·

• Add standalone TdsKVBackend shell and path mapper for TDS storage.

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

  ·
  1

• Implement in-place host buffer tiling and parameterize tiling unit tests

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

  ·

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

  #1172·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·

  ·

• Add sampler-side bundle pull engine to WeightSynchronizerBase

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

  ·

• Wire SwarmService and bundle pull commands to WeightSynchronizerListener

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

  ·

• Add PullBuffer to BlockTransport and RaidenManagerBase

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

  ·

• Added the controller v3 logical types and work-unit entity registry.

  #1162·copybara-service[bot] opened Oct 4, 2026Bot·

  ·

• Add SwarmService for bundle pull coordination

  #1161·copybara-service[bot] opened Oct 4, 2026Bot·

  ·

• Split H2dLayer into TileLayer + raw H2dLayer so deferred H2D copies pre-tiled data

  #1160·copybara-service[bot] opened Oct 4, 2026Bot·

  ·

• Add V3 bundle pull and swarm messages to raiden_service.proto

  #1159·copybara-service[bot] opened Oct 4, 2026Bot·

  ·

• Migrate callers of internal torch.tpu device APIs to public torch.accelerator APIs.

  #1155·copybara-service[bot] opened Oct 3, 2026Bot·

  ·

• Migrate callers of internal torch.tpu device APIs to public torch.accelerator APIs.

  #1154·copybara-service[bot] opened Oct 3, 2026Bot·

  ·

• Migrate callers of internal torch.tpu device APIs to public torch.accelerator APIs.

  #1153·copybara-service[bot] opened Oct 3, 2026Bot·

  ·

• [jax multi-version] Mirror jaxlib's PyArray layout without its headers; build OSS with visibility checks on.

  #1151·copybara-service[bot] opened Oct 2, 2026Bot·

  ·

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

  #1149·copybara-service[bot] opened Oct 2, 2026Bot·

  ·

• Put the JAX version in the jax wheel build tag.

  #1148·copybara-service[bot] opened Oct 2, 2026Bot·

  ·

• Add fault injection hooks for tpu_sync transport and receive paths:

  #1138·copybara-service[bot] opened Oct 1, 2026Bot·

  ·
  1

• Internal change only

  #1135·copybara-service[bot] opened Oct 1, 2026Bot·

  ·