-
Notifications
You must be signed in to change notification settings - Fork 28
Pull requests: google/tpu-sync
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Implement ShmCollector to scan, validate, and aggregate telemetry metrics across all active worker memory-mapped files in /dev/shm.
#815
opened Aug 29, 2026 by
copybara-service
Bot
Loading…
Optimized weights sync transport by multi-NUMA.
#814
opened Aug 29, 2026 by
copybara-service
Bot
Loading…
Integrate Raiden with Cerebro via UserDefinedOps.
#812
opened Aug 29, 2026 by
copybara-service
Bot
Loading…
Refactor the script and added 3 test cases: test_non_contiguous_blocks, test_host_reordering, test_large_complex_non_contiguous_and_reorder
#810
opened Aug 29, 2026 by
copybara-service
Bot
Loading…
Cut fixed per-request costs out of the reshard control plane
#808
opened Aug 28, 2026 by
yinlin09
Collaborator
Loading…
Add multi-dimensional label buffering and metric routing to BufferedMetricsExporter.
#800
opened Aug 27, 2026 by
copybara-service
Bot
Loading…
Implement PrometheusShmExporter to expose aggregated shared memory metrics over HTTP on port 9090.
#799
opened Aug 27, 2026 by
copybara-service
Bot
Loading…
Unify JAX WeightSynchronizer API across Pathways FFI and native backends
#794
opened Aug 27, 2026 by
copybara-service
Bot
Loading…
Add multi-dimensional telemetry labels support to TPU Raiden.
#789
opened Aug 27, 2026 by
copybara-service
Bot
Loading…
Implement low-overhead, memory-mapped ShmWriter for recording TPU Raiden telemetry metrics directly into POSIX shared memory (/dev/shm).
#782
opened Aug 26, 2026 by
copybara-service
Bot
Loading…
Fix compilation errors in KVCacheStoreMock.
#775
opened Aug 24, 2026 by
copybara-service
Bot
Loading…
Fix PyTorch missing visibility compilation failures
#774
opened Aug 24, 2026 by
copybara-service
Bot
Loading…
[tpu_raiden] Hardening: async recovery dispatch, result snapshotting, clock decoupling, and teardown cancellation
#773
opened Aug 24, 2026 by
copybara-service
Bot
Loading…
Fix compilation errors and test failures across tpu_sync targets
#772
opened Aug 24, 2026 by
copybara-service
Bot
Loading…
Add support for persistent storage backend offloading and recalling in TPU Raiden
#768
opened Aug 24, 2026 by
copybara-service
Bot
Loading…
raw_transfer: compile across the XLA PjRtRawBufferInterface->CommonPjRtRawBuffer rename
#732
opened Aug 20, 2026 by
lokic233
Loading…
Adopt the caller's device block when re-inserting an HBM-only entry.
#728
opened Aug 20, 2026 by
copybara-service
Bot
Loading…
Harden legacy UUID KV transfer leases and teardown
pull ready
#723
opened Aug 19, 2026 by
wangyang59
Loading…
Demote cold KV blocks to placement targets when free blocks run low.
#712
opened Aug 18, 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…
Previous Next
ProTip!
Filter pull requests by the default branch with base:main.