Skip to content

Pull requests: google/tunix

New pull request New
187 Open 1,323 Closed
187 Open 1,323 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

Add audio support to Gemma4 sampler.
#1622 opened Jun 25, 2026 by copybara-service Bot Loading…
@tianshub
Bump version from 0.1.7 to 0.1.8
#1621 opened Jun 25, 2026 by tejaswin432 Collaborator Loading…
@tianshub
[Tunix] Add multi-modal (vision) support to Gemma 4.
#1608 opened Jun 21, 2026 by msghik Loading…
1
@jiangyangmu
Consolidate CLI RL logic
#1602 opened Jun 17, 2026 by yatlas20098 Collaborator Loading…
6 tasks done
@abheesht17
cli changes
#1598 opened Jun 16, 2026 by copybara-service Bot Loading…
@abheesht17
Add chunked prefill with prefix KV cache reuse to Tunix Gemma 4.
#1597 opened Jun 16, 2026 by copybara-service Bot Loading…
@abheesht17
1
Internal
#1596 opened Jun 16, 2026 by copybara-service Bot Loading…
@abheesht17
1
Introduce SubBatchCheckpointManager interface.
#1594 opened Jun 15, 2026 by copybara-service Bot Loading…
@abheesht17
Add support for keep_every_nth_step in checkpointing options.
#1593 opened Jun 15, 2026 by copybara-service Bot Loading…
@abheesht17
Rename agentic GSM8K examples to a dedicated math_gsm8k name.
#1591 opened Jun 15, 2026 by copybara-service Bot Loading…
@abheesht17
[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…
@abheesht17
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…
@abheesht17
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…
@sizhit2
Add text-backbone support for Gemma 4 12B and 12B-it
#1570 opened Jun 9, 2026 by deep-diver Loading…
@sizhit2
1
Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#1567 opened Jun 5, 2026 by copybara-service Bot Loading…
@sizhit2
Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#1565 opened Jun 5, 2026 by copybara-service Bot Loading…
@sizhit2
fix(rl): accept numpy ints and reject bool in is_positive_integer
#1552 opened May 29, 2026 by kitrakrev Loading…
2 tasks done
@hgao327
2
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…
1
@hgao327
[Tunix] Add multi-modal (vision) support to Gemma 4.
#1545 opened May 27, 2026 by msghik Loading…
1
@lc5211
3
Add Trackio rollout trace logging
#1534 opened May 22, 2026 by abidlabs Loading…
@lc5211
2
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…
@wang2yn84
shard perfetto trace writes and bound Timeline memory
#1529 opened May 18, 2026 by copybara-service Bot Loading…
@wang2yn84
Add auto-seeding for rollouts.
#1526 opened May 15, 2026 by copybara-service Bot Loading…
@wang2yn84
update qwen3 scripts to use yaml configs
#1520 opened May 14, 2026 by copybara-service Bot Loading…
@wang2yn84
Inject unified Tunix and vLLM environment fingerprinting metadata
#1517 opened May 14, 2026 by copybara-service Bot Loading…
@wang2yn84
ProTip! no:milestone will show everything without a milestone.