Skip to content

Pull requests: google/tpu-raiden

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

Support 1D buffer cache transfers by explicitly providing block size when constructing KV cache manager.
#271 opened Jul 6, 2026 by copybara-service Bot Loading…
Support 1D buffer cache transfers by explicitly providing block size when constructing KV cache manager.
#270 opened Jul 6, 2026 by copybara-service Bot Loading…
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…
Replace RaidenId with RaidenBlockID, add ReleaseAndDelete, and store RaidenId in KVCacheStore.
#262 opened Jul 6, 2026 by copybara-service Bot Loading…
Implement global registry lookup fallback for KVCacheStore.
#261 opened Jul 6, 2026 by copybara-service Bot Loading…
Coalesce contiguous copies on d2h/h2d.
#242 opened Jul 1, 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…
Support per-layer buffer sizes in KVCacheManagerBase for hybrid models.
#219 opened Jun 29, 2026 by copybara-service Bot Loading…
1
Commit 1: Category 5 - GKE Infra
#213 opened Jun 28, 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! Exclude everything labeled bug with -label:bug.