-
Notifications
You must be signed in to change notification settings - Fork 335
Pull requests: google/tunix
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
[Weight sync Raiden 3/N] Add the raiden sampler adapter and trainer staging
#1935
opened Aug 20, 2026 by
copybara-service
Bot
Loading…
[Weight Sync Orchestrator 2/N] Route sync_weights through the weight sync coordinator
#1933
opened Aug 19, 2026 by
copybara-service
Bot
Loading…
Complete the weight sync round contract on workers
#1932
opened Aug 19, 2026 by
copybara-service
Bot
Loading…
Refactor rollout dispatch in DistributedRLEngine with group_size expansion and typed GenerationArgs.
#1929
opened Aug 19, 2026 by
copybara-service
Bot
Loading…
Allow client keep alive pings in worker gRPC servers
#1928
opened Aug 19, 2026 by
copybara-service
Bot
Loading…
Update pinned commits for vllm and tpu-inference
#1927
opened Aug 19, 2026 by
niting
Contributor
Loading…
Parse chat messages for string-prompt samplers
#1926
opened Aug 19, 2026 by
copybara-service
Bot
Loading…
Fix rollout mesh axis names for gemma sharding
#1925
opened Aug 19, 2026 by
copybara-service
Bot
Loading…
Add the raiden worker layer for weight sync.
#1923
opened Aug 19, 2026 by
copybara-service
Bot
Loading…
Fix Gemma 4 splash attention divisibility check.
#1921
opened Aug 18, 2026 by
copybara-service
Bot
Loading…
fix: version mismatch in flax and jax in tpu test
#1913
opened Aug 17, 2026 by
copybara-service
Bot
Loading…
feat: Add VllmSamplerAdapter wrapping tpu-inference RLVllmSampler
#1912
opened Aug 17, 2026 by
khatwanimohit
Collaborator
Loading…
6 tasks done
Add experimental vanilla sampler v2 for LLM generation
#1908
opened Aug 17, 2026 by
copybara-service
Bot
Loading…
feat(experimental): integrate TPU-Raiden weight synchronization
#1907
opened Aug 17, 2026 by
khatwanimohit
Collaborator
•
Draft
6 tasks
Support chunked prefill with prefix KV cache reuse in Tunix Gemma 4.
#1906
opened Aug 17, 2026 by
copybara-service
Bot
Loading…
Add optional process-pool evaluation of reward functions
#1904
opened Aug 17, 2026 by
gutianyu-google
•
Draft
6 tasks done
[Weight sync source 2/N] Add a synchronous driver for the coordinator
#1899
opened Aug 16, 2026 by
copybara-service
Bot
Loading…
Implement cross-layer KV cache sharing lifecycle in Tunix Gemma 4.
#1896
opened Aug 15, 2026 by
copybara-service
Bot
Loading…
[tunix] Add target-aligned diffusion SFT adapter
#1891
opened Aug 14, 2026 by
ethannnnnn
Contributor
Loading…
Modernize Pallas Splash attention kernel and define eager fallback routing in Tunix Gemma 4.
#1890
opened Aug 14, 2026 by
copybara-service
Bot
Loading…
[Tunix] Add end-to-end data lineage tracking and unified identifier semantics for Tunix RL.
#1887
opened Aug 14, 2026 by
copybara-service
Bot
Loading…
Previous Next
ProTip!
Type g i on any issue or pull request to go back to the issue listing page.