-
Notifications
You must be signed in to change notification settings - Fork 22
Pull requests: google/tpu-sync
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Optimize
bfloat16 and nested sub-tile packed tiling in weight synchronizer.
#685
opened Aug 17, 2026 by
copybara-service
Bot
Loading…
[2/N][kv_cache_store api] settle a remote write exactly once, and never leave one behind
#684
opened Aug 17, 2026 by
copybara-service
Bot
Loading…
[Weight sync source 2/N] Add a synchronous driver for the coordinator
#683
opened Aug 16, 2026 by
copybara-service
Bot
Loading…
[Weight sync source 1/N] Keep Trainer Worker staged until release
#682
opened Aug 16, 2026 by
copybara-service
Bot
Loading…
Complete PJRT copy futures via event callbacks instead of awaiter threads
#681
opened Aug 16, 2026 by
copybara-service
Bot
Loading…
kv_cache: report abandoned pushes as failed, not sent
#679
opened Aug 16, 2026 by
Oseltamivir
Loading…
[kv_cache] Expose kv_cache_pull duration as a metric
#678
opened Aug 16, 2026 by
copybara-service
Bot
Loading…
[kv_transfer] Add transfer latency metrics for H2D and D2H
#676
opened Aug 16, 2026 by
copybara-service
Bot
Loading…
Keep the peer store client across WriteRemote refusals.
#674
opened Aug 16, 2026 by
copybara-service
Bot
Loading…
Wire placement and monitor configuration into the store entry points.
#673
opened Aug 16, 2026 by
copybara-service
Bot
Loading…
weight-sync(controller): fail-loud full source/destination coverage guard
#667
opened Aug 15, 2026 by
lokic233
Loading…
weight-sync(jax): fail-loud runtime-derived tile-representability guard
#666
opened Aug 15, 2026 by
lokic233
Loading…
weight-sync: fast-path host detile for the bf16 two-tile TPU layout
#665
opened Aug 15, 2026 by
lokic233
Loading…
weight-sync: use PJRT event-completion path for D2H (fix silent rank-1 corruption)
#664
opened Aug 15, 2026 by
lokic233
Loading…
[tpu_raiden]Add PeregrineControlService TPU Raiden transport
#659
opened Aug 15, 2026 by
copybara-service
Bot
Loading…
Clean up internal references, codenames, and agent configurations from exported files.
#657
opened Aug 14, 2026 by
copybara-service
Bot
Loading…
Optimize KVCacheStore polling by updating loaded block handles in-place.
#656
opened Aug 14, 2026 by
copybara-service
Bot
Loading…
Demote cold KV blocks to placement targets when free blocks run low.
#655
opened Aug 14, 2026 by
copybara-service
Bot
Loading…
[tpu_raiden] High-Level CL sketch to refactor block transport to Post/Poll API to enable integration with Peregrine API in future iterations.
#643
opened Aug 14, 2026 by
copybara-service
Bot
Loading…
tpu_raiden: Fix copy.bara.sky exclude glob path for metrics_api
#637
opened Aug 13, 2026 by
copybara-service
Bot
Loading…
tpu_raiden: Unblock Copybara pipeline sequentially stuck on historical static copyright notices
#636
opened Aug 13, 2026 by
copybara-service
Bot
Loading…
JAX: pool admission + control-plane listener parity with torch
#634
opened Aug 13, 2026 by
copybara-service
Bot
Loading…
[tpu_raiden] Make Prometheus telemetry optional and disabled by default.
#627
opened Aug 13, 2026 by
copybara-service
Bot
Loading…
Previous Next
ProTip!
Find all pull requests that aren't related to any open issues with -linked:issue.