Pull requests: google/tunix
Pull requests list
clean up rest of the rl_program tests to use the helper functions
#1959
opened Aug 22, 2026 by
copybara-service
Bot
Loading…
Add TODO, construct the WeightSynchronizer inside the trainer
#1958
opened Aug 22, 2026 by
copybara-service
Bot
Loading…
Stop losing queued trajectory writes when a process exits.
#1954
opened Aug 21, 2026 by
copybara-service
Bot
Loading…
Implement split attention optimization with LSE merge for chunked prefill in Tunix Gemma 4.
#1950
opened Aug 21, 2026 by
copybara-service
Bot
Loading…
[Tunix] Add load-aware routing and health monitoring to the distributed RL engine.
#1946
opened Aug 21, 2026 by
copybara-service
Bot
Loading…
[Tunix] Refactor PaddedBatchAssembler and RLTrainerPayload to support separate prompt and completion padding.
#1944
opened Aug 20, 2026 by
copybara-service
Bot
Loading…
[Tunix] add tokamax splash attention as an option in addition to jax splash attention.
#1943
opened Aug 20, 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…
Add optional process-pool evaluation of reward functions
#1904
opened Aug 17, 2026 by
gutianyu-google
•
Draft
6 tasks done
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…
[Tunix]Extract TrainingConfig to a standalone module and expose it at the package level.
#1885
opened Aug 13, 2026 by
copybara-service
Bot
Loading…
Enforce zero-ignore static typing rigor across ShardingConfig and support activation checkpointing in Tunix Gemma 4.
#1877
opened Aug 12, 2026 by
copybara-service
Bot
Loading…
ProTip!
Follow long discussions with comments:>50.