Pull requests: google/tunix
Pull requests list
Consolidate CLI RL logic
#1602
opened Jun 17, 2026 by
yatlas20098
Collaborator
Loading…
6 tasks done
Add chunked prefill with prefix KV cache reuse to Tunix Gemma 4.
#1597
opened Jun 16, 2026 by
copybara-service
Bot
Loading…
Introduce SubBatchCheckpointManager interface.
#1594
opened Jun 15, 2026 by
copybara-service
Bot
Loading…
Add support for
keep_every_nth_step in checkpointing options.
#1593
opened Jun 15, 2026 by
copybara-service
Bot
Loading…
Rename agentic GSM8K examples to a dedicated math_gsm8k name.
#1591
opened Jun 15, 2026 by
copybara-service
Bot
Loading…
[Tunix] Don't skip the special token from the vllm responses in agentic training otherwise, we will need manual processing to previous token shifts.
#1588
opened Jun 15, 2026 by
copybara-service
Bot
Loading…
Update Checkpointing Options construction through CLI configuration to move from Orbax V0 CheckpointManagerOptions to Tunix's CheckpointOptions.
#1584
opened Jun 12, 2026 by
copybara-service
Bot
Loading…
Migrate Tunix's SFT Checkpoint Manager usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#1571
opened Jun 9, 2026 by
copybara-service
Bot
Loading…
Add text-backbone support for Gemma 4 12B and 12B-it
#1570
opened Jun 9, 2026 by
deep-diver
Loading…
Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#1567
opened Jun 5, 2026 by
copybara-service
Bot
Loading…
Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#1565
opened Jun 5, 2026 by
copybara-service
Bot
Loading…
fix(rl): accept numpy ints and reject bool in is_positive_integer
#1552
opened May 29, 2026 by
kitrakrev
Loading…
2 tasks done
Fix TypeError when passing auto-derived segment_ids to models that don't accept it (Gemma/Llama)
#1547
opened May 28, 2026 by
msghik
Loading…
Update sharding for KV heads in Gemma4 4B model to allow for tp_size=4
#1530
opened May 19, 2026 by
copybara-service
Bot
Loading…
shard perfetto trace writes and bound Timeline memory
#1529
opened May 18, 2026 by
copybara-service
Bot
Loading…
Inject unified Tunix and vLLM environment fingerprinting metadata
#1517
opened May 14, 2026 by
copybara-service
Bot
Loading…
Add RL smoke tests to Tunix TPU CI pipeline
#1516
opened May 14, 2026 by
copybara-service
Bot
Loading…
ProTip!
Filter pull requests by the default branch with base:main.