-
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
Add buffer occupancy telemetry to TPU Raiden KV cache manager.
#693
opened Aug 17, 2026 by
copybara-service
Bot
Loading…
[4/N][kv_cache_store api] Application-level lookup() pins what it returns
#692
opened Aug 17, 2026 by
copybara-service
Bot
Loading…
KV cache evict e2e test and H2hRead dst_offsets bug fix.
#691
opened Aug 17, 2026 by
copybara-service
Bot
Loading…
[3/N][kv_cache_store api] lookup() no longer consults the global registry by default
#690
opened Aug 17, 2026 by
copybara-service
Bot
Loading…
[Weight sync destination 4/N] Complete RolloutWorker's destination entry points
#689
opened Aug 17, 2026 by
copybara-service
Bot
Loading…
[Weight sync destination 3/N] Close rollout admission during weight sync rounds
#688
opened Aug 17, 2026 by
copybara-service
Bot
Loading…
[Weight sync destination 2/N] Add a TrafficController for rollout worker admission and state (credit: noghabi)
#687
opened Aug 17, 2026 by
copybara-service
Bot
Loading…
[Weight sync destination 1/N] Expose sampler weight sync metadata through RolloutManager
#686
opened Aug 17, 2026 by
copybara-service
Bot
Loading…
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…
Previous Next
ProTip!
Updated in the last three days: updated:>2026-08-13.