Skip to content

Pull requests: google/tpu-raiden

New pull request New
33 Open 205 Closed
33 Open 205 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

The eviction-triggered D2h offload reads KV blocks straight from the
#235 opened Jun 30, 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…
Implement BDF/MTU-based TPU network topology and NUMA mapping discovery
#231 opened Jun 30, 2026 by copybara-service Bot Loading…
Update change to my local benchmark testing branch
#221 opened Jun 29, 2026 by haochengd-google Collaborator Loading…
1
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…
Import CL 939200938 with GKE DRANET adaptations and control NIC exclusion.
#212 opened Jun 28, 2026 by copybara-service Bot Loading…
Implemented layered infrastructure for resharding KV cache with push model.
#211 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
Fix physical data corruption in tpu_raiden by implementing CPU Cache Displacement.
#190 opened Jun 25, 2026 by copybara-service Bot Loading…
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…
feat(core): Enhance HostMemoryAllocator to support optional shared memory allocation
#171 opened Jun 23, 2026 by copybara-service Bot Loading…
1
Add file write/read verification test to HostMemoryAllocatorTest
#170 opened Jun 23, 2026 by copybara-service Bot Loading…
1
Add MT contention benchmark for Raiden
#166 opened Jun 22, 2026 by copybara-service Bot Loading…
Refactor TPU Raiden core to disallow C++ exceptions.
#163 opened Jun 20, 2026 by copybara-service Bot Loading…
Retry WriteBlockDirect on stale-connection failures
#159 opened Jun 20, 2026 by copybara-service Bot Loading…
Fix multi-process DmaMap out-of-bounds abort and refactor host memory allocator fallback
#157 opened Jun 20, 2026 by copybara-service Bot Loading…
ProTip! Mix and match filters to narrow down what you’re looking for.