Pull requests: google/tunix
Pull requests list
Add cli for frozenlake gemma4-e2b-it recipe
#1626
opened Jun 26, 2026 by
lc5211
Collaborator
Loading…
6 tasks done
WIP: Add audio support for tunix/models/gemma4
#1625
opened Jun 26, 2026 by
copybara-service
Bot
Loading…
11 of 13 tasks
Fix gemma4 vision encoder Orbax parameter mapping and add
text_only flag to checkpoint loader.
#1624
opened Jun 26, 2026 by
copybara-service
Bot
Loading…
Fix bug in trainer loop and optimize parallel rollout
#1623
opened Jun 26, 2026 by
AntonyMei
Collaborator
Loading…
6 tasks done
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…
ProTip!
Type g p on any issue or pull request to go back to the pull request listing page.