Skip to content

Pull requests: google/tpu-raiden

New pull request New
43 Open 300 Closed
43 Open 300 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Enabled multi-host resharding support in weight synchronizer
#343 opened Jul 16, 2026 by copybara-service Bot Loading…
Implement Evict, Proactive Eviction, and LRU Candidate List in KVCacheStore.
#342 opened Jul 16, 2026 by copybara-service Bot Loading…
Presubmit Critical Path Gating (GitHub Workflow)
#341 opened Jul 16, 2026 by haochengd-google Collaborator Loading…
2
Implement asynchronous Save and Load in KVCacheStore, and fix JAX OSS build.
#339 opened Jul 15, 2026 by copybara-service Bot Loading…
Implemented GetTransferStatus to allow clients to query the status of a transfer from the Raiden Controller.
#337 opened Jul 15, 2026 by copybara-service Bot Loading…
[tpu_raiden]
#336 opened Jul 15, 2026 by copybara-service Bot Loading…
1
Fix register_active_plan nanobind type to accept bytes instead of str.
#335 opened Jul 15, 2026 by copybara-service Bot Loading…
1
[tpu_raiden] improve transport library unit tests
#334 opened Jul 15, 2026 by copybara-service Bot Loading…
1
Retain legacy TransferBuffers APIs and implement H2hRead / H2hWrite routing.
#332 opened Jul 15, 2026 by copybara-service Bot Loading…
Update personal agent configurations for tpu_raiden migration.
#330 opened Jul 15, 2026 by copybara-service Bot Loading…
feat: implement Raiden OTEL design and telemetry
#321 opened Jul 14, 2026 by copybara-service Bot Loading…
1
Test PR Gating - DO NOT MERGE
#296 opened Jul 10, 2026 by haochengd-google Collaborator Loading…
Add pre-submit gating test (H2D/D2H/H2H)
#290 opened Jul 9, 2026 by haochengd-google Collaborator Loading…
2
Add Global Prefix Caching design documentation for TPU Raiden.
#279 opened Jul 8, 2026 by copybara-service Bot Loading…
Add KVCacheManager.read_host_bytes and write_host_bytes: length-bounded
#278 opened Jul 8, 2026 by copybara-service Bot Loading…
1
Expose the optional dst_block_ids parameter in the JAX KVCacheManager.h2h_write Python binding to align with the underlying C++ signature.
#274 opened Jul 7, 2026 by copybara-service Bot Loading…
1
Implement remote-fetch destination auto-allocation.
#269 opened Jul 6, 2026 by copybara-service Bot Loading…
Fix OSS build and test execution issues.
#268 opened Jul 6, 2026 by copybara-service Bot Loading…
Implement KVCacheStore::Evict (DRAM Cleanup) API.
#267 opened Jul 6, 2026 by copybara-service Bot Loading…
Implement KVCacheStore::Save (D2H) with write-through to global registry.
#266 opened Jul 6, 2026 by copybara-service Bot Loading…
Implement Host-to-Device Load API in KVCacheStore.
#265 opened Jul 6, 2026 by copybara-service Bot Loading…
Implement async global registry write-through on remote fetch completion in TPU Raiden KV Cache.
#264 opened Jul 6, 2026 by copybara-service Bot Loading…
Implement Remote Fetch Batching and Python Bindings
#263 opened Jul 6, 2026 by copybara-service Bot Loading…
Remove redundant jax_kv_cache_manager_perf_test_gf.
#233 opened Jun 30, 2026 by copybara-service Bot Loading…
Add dynamic KVCache and metadata resolution to support Qwen3 serving.
#208 opened Jun 27, 2026 by copybara-service Bot Loading…
ProTip! no:milestone will show everything without a milestone.