Skip to content

Pull requests: google/tpu-raiden

New pull request New
14 Open 135 Closed
14 Open 135 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

Refactor KVCacheManagerWithTransfer control plane to use transport commands API, removing separate control server port and thread.
#149 opened Jun 18, 2026 by copybara-service Bot Loading…
Consolidate internal directories into unified tpu_raiden/ namespace
#148 opened Jun 18, 2026 by copybara-service Bot Loading…
Fix dangling pointer for PyTorch view tensors in async operations by keeping materialized PjRtBuffer alive via DeviceBufferRef.
#146 opened Jun 18, 2026 by copybara-service Bot Loading…
1
Implement C++ NUMA thread pinning and socket-buffer affinity for JAX H2H transfers.
#145 opened Jun 18, 2026 by copybara-service Bot Loading…
Refactor coordination proto and helper for multi-NIC structured endpoints
#144 opened Jun 18, 2026 by copybara-service Bot Loading…
Add CPU-only GitHub Actions CI (presubmit + nightly + lint)
#143 opened Jun 18, 2026 by yinlin09 Collaborator Loading…
[Experimental] Raiden H2H setup and benchmark scripts.
#139 opened Jun 17, 2026 by copybara-service Bot Loading…
Implemented the first draft of resharding for KV cache on torch tensors, using Raiden controller and push model.
#136 opened Jun 17, 2026 by copybara-service Bot Loading…
Implement multi-NIC C++ transport with PCIe-NUMA auto-discovery and FFI registry.
#135 opened Jun 17, 2026 by copybara-service Bot Loading…
Implement multi-NIC transport socket binding and GKE /sys mount to restore dual-NIC performance.
#124 opened Jun 15, 2026 by copybara-service Bot Loading…
Refactor build to link against pre-compiled JAX and PyTorch wheels.
#107 opened Jun 12, 2026 by copybara-service Bot Loading…
[peregrine] extend transport api for posting pipelined requests
#91 opened Jun 11, 2026 by copybara-service Bot Loading…
2
Harden Kokoro presubmit, isolate GCS cache, and add an internal wheel sync script.
#89 opened Jun 11, 2026 by copybara-service Bot Loading…
Use status macro to fix build failure
#56 opened Jun 4, 2026 by copybara-service Bot Loading…
ProTip! no:milestone will show everything without a milestone.