-
Notifications
You must be signed in to change notification settings - Fork 31
Pull requests: google/tpu-sync
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Adopt the torch_tpu tensor_buffer API and drop the nobuilder build exclusions.
#877
opened Sep 7, 2026 by
copybara-service
Bot
Loading…
Integrate PrometheusShmExporter into the RaidenMetricStore facade and add multi-process end-to-end stress tests.
#874
opened Sep 7, 2026 by
copybara-service
Bot
Loading…
Fix control handshake MSAN flake and socket teardown in KVCacheManagerWithTransfer.
#873
opened Sep 7, 2026 by
copybara-service
Bot
Loading…
Add
torch_dma_kv_cache_benchmark.py under examples/microbenchmarks/ providing PyTorch parity with jax_dma_kv_cache_benchmark.py:
#871
opened Sep 6, 2026 by
copybara-service
Bot
Loading…
Instrument weight synchronization components with coarse-grained
RAIDEN_TRACE annotations.
#860
opened Sep 5, 2026 by
copybara-service
Bot
Loading…
Implement ShmCollector to scan, validate, and aggregate telemetry metrics across all active worker memory-mapped files in /dev/shm.
#852
opened Sep 4, 2026 by
copybara-service
Bot
Loading…
Integrate WeightSynchronizationWorkerService gRPC mode into WeightSynchronizer and RaidenController.
#849
opened Sep 4, 2026 by
copybara-service
Bot
Loading…
Add the LLVM apt source only when the image lacks it
pull ready
#836
opened Sep 1, 2026 by
yinlin09
Collaborator
Loading…
Expand PyTorch KVCacheStore test suite to full parity with JAX:
#830
opened Sep 1, 2026 by
copybara-service
Bot
Loading…
Expand PyTorch WeightSynchronizer test suite to reach parity with JAX:
#828
opened Aug 31, 2026 by
copybara-service
Bot
Loading…
Update TorchTPU KVCacheManager transfer unit test suite and add permutation test cases to reach full parity with JAX:
#824
opened Aug 31, 2026 by
copybara-service
Bot
Loading…
Define BaseShmExporter for multi-process shared-memory telemetry exporters.
#820
opened Aug 30, 2026 by
copybara-service
Bot
Loading…
Cut fixed per-request costs out of the reshard control plane
#816
opened Aug 29, 2026 by
copybara-service
Bot
Loading…
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…
Integrate Raiden with Cerebro via UserDefinedOps.
#812
opened Aug 29, 2026 by
copybara-service
Bot
Loading…
Refactor KVCacheManager unit test suite and add transfer test cases:
#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…
Implement PrometheusShmExporter to expose aggregated shared memory metrics via a Prometheus HTTP endpoint (/metrics) and text snapshots.
#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…
Fix compilation errors in KVCacheStoreMock.
#775
opened Aug 24, 2026 by
copybara-service
Bot
Loading…
Previous Next
ProTip!
Find all pull requests that aren't related to any open issues with -linked:issue.