-
Notifications
You must be signed in to change notification settings - Fork 21
Pull requests: google/tpu-sync
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
[kv_cache] Expose kv_cache_pull duration as a metric
#678
opened Aug 16, 2026 by
copybara-service
Bot
Loading…
Add per-tag dst_skip_bytes clip for prefix-aware suffix reshard.
#677
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…
Respected s
kip_tiling in RaidenController chunk generation for linear host transfers.
#675
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…
Batch byte transfer telemetry metrics per stream in BlockTransport.
#672
opened Aug 15, 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…
examples: fix stale tpu_raiden paths in microbenchmarks
#663
opened Aug 15, 2026 by
Sahil170595
Loading…
ci: point benchmark registry_file at tpu_sync/ after directory migration
#662
opened Aug 15, 2026 by
Sahil170595
Loading…
Optimized TPU weight tiling and detiling routines for padded tensor shapes.
#661
opened Aug 15, 2026 by
copybara-service
Bot
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…
Migrate third_party/tpu_raiden/tpu_raiden/weight_sync to third_party/tpu_raiden/tpu_sync/weight_sync.
#628
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!
Type g p on any issue or pull request to go back to the pull request listing page.