Skip to content

Pull requests: google/tpu-raiden

New pull request New
34 Open 264 Closed
34 Open 264 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

[tpu-raiden] check all sockets are blocking
#298 opened Jul 11, 2026 by copybara-service Bot Loading…
1
[tpu-raiden] refactor socket send/recv functions out of RawBufferTransport
#297 opened Jul 11, 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…
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…
Create single host Raiden KV offloading example scripts.
#207 opened Jun 27, 2026 by copybara-service Bot Loading…
Fix sockets failures in IPv6-only environments (e.g., Borg runinvm)
#206 opened Jun 26, 2026 by copybara-service Bot Loading…
1
[TEMP] Inject temporary Docker build into presubmit
#199 opened Jun 26, 2026 by copybara-service Bot Loading…
1
Create custom Docker image and Kokoro job to build it
#198 opened Jun 26, 2026 by copybara-service Bot Loading…
1
Add clang and llvm to the ml-build container to support building XLA and related tooling out-of-the-box.
#193 opened Jun 25, 2026 by copybara-service Bot Loading…
1
Add Dockerfile for BAP ML-Actions and configure benchmark registry to test it
#192 opened Jun 25, 2026 by copybara-service Bot Loading…
1
Support IPv6-only environments in GetLocalHostNicAddresses.
#189 opened Jun 25, 2026 by copybara-service Bot Loading…
Support IPv6-only environments in GetLocalHostNicAddresses.
#188 opened Jun 25, 2026 by copybara-service Bot Loading…
Bump JAX to 0.10.1, jaxlib to 0.10.1, and libtpu to 0.0.41.
#186 opened Jun 25, 2026 by copybara-service Bot Loading…
ProTip! Mix and match filters to narrow down what you’re looking for.