Skip to content

Pull requests: google/tunix

New pull request New
199 Open 1,327 Closed
199 Open 1,327 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

Refactor checkpointing to use Orbax v1.
#1638 opened Jul 1, 2026 by copybara-service Bot Loading…
@tianshub
Remove degen groups.
#1637 opened Jul 1, 2026 by wang2yn84 Collaborator Loading…
6 tasks
@tianshub
Fix garbage collection of pending cancellation_task in AgenticRLLearner
#1636 opened Jun 30, 2026 by AntonyMei Collaborator Loading…
6 tasks done
@tianshub
Fix degenerate group masking for multi-group batching in GRPO learner
#1635 opened Jun 30, 2026 by AntonyMei Collaborator Loading…
6 tasks done
@tianshub
Verify model supports segment-ids
#1632 opened Jun 30, 2026 by yatlas20098 Collaborator Loading…
6 tasks done
@tianshub
1
Stop sampler decode one step earlier
#1631 opened Jun 28, 2026 by fallintoplace Loading…
@tianshub
Fix beam search test helper indexing
#1630 opened Jun 28, 2026 by fallintoplace Loading…
@tianshub
Fix example max_steps calculations
#1629 opened Jun 28, 2026 by fallintoplace Loading…
@tianshub
- Create BasePipeline in tunix/cli/base_rl_main.py to consolidate RL logic
#1628 opened Jun 28, 2026 by copybara-service Bot Loading…
@tianshub
1
Guard duplicate-key check and write in safetensors loader with a lock
#1627 opened Jun 27, 2026 by AMR5210 Loading…
6 tasks done
1
@tianshub
1
Add cli for frozenlake gemma4-e2b-it recipe
#1626 opened Jun 26, 2026 by lc5211 Collaborator Loading…
6 tasks done
@tianshub
Fix bug in trainer loop and optimize parallel rollout
#1623 opened Jun 26, 2026 by AntonyMei Collaborator Loading…
6 tasks done
@tianshub
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 • Draft
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
ProTip! Mix and match filters to narrow down what you’re looking for.