Skip to content

Pull requests: google/tpu-raiden

New pull request New
36 Open 194 Closed
36 Open 194 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

Implement unlock_blocks in JAX KVCacheManager.
#230 opened Jun 30, 2026 by copybara-service Bot Loading…
Refactor MetricsCollector to dynamically monitor active data interfaces
#229 opened Jun 30, 2026 by copybara-service Bot Loading…
Remove gcloud auth login from TPU Raiden pip install instructions
#228 opened Jun 30, 2026 by copybara-service Bot Loading…
1
Removed unnecessary construction of local_address_.
#224 opened Jun 29, 2026 by copybara-service Bot Loading…
Remove unused dependencies from jax:pjrt_client_extension.
#223 opened Jun 29, 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…
ProTip! Adding no:label will show everything without a label.