Skip to content

Pull requests: google/tunix

New pull request New
197 Open 1,349 Closed
197 Open 1,349 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

[Seq Packing] Unified sequence packing: packed logprob + pack-first (agentic GRPO)
#1659 opened Jul 10, 2026 by yuxuandexter Collaborator Loading…
@lc5211
1
Update few type annotations.
#1658 opened Jul 10, 2026 by copybara-service Bot Loading…
@lc5211
Add options for using rollout logprobs and token-based importance sampling.
#1657 opened Jul 9, 2026 by copybara-service Bot Loading…
@lc5211
feat: deployment scaffolding initial change.
#1656 opened Jul 9, 2026 by copybara-service Bot Loading…
@lc5211
[Seq Packing] support fsdp packing in sequence packing
#1655 opened Jul 9, 2026 by copybara-service Bot Loading…
@lc5211
1
Update deepswe example imports for r2egym.
#1654 opened Jul 8, 2026 by copybara-service Bot Loading…
@wang2yn84
[Seq Packing] stream based gradient accumulation
#1650 opened Jul 7, 2026 by copybara-service Bot Loading…
@wang2yn84
1
Adding type suppressions for pyrefly
#1649 opened Jul 7, 2026 by copybara-service Bot Loading…
@wang2yn84
Fix dependency errors and TPU OOM crash in Gemma DPO notebook
#1648 opened Jul 7, 2026 by selamw1 Collaborator Loading…
@wang2yn84
prevent 403 error on naming test
#1647 opened Jul 7, 2026 by copybara-service Bot Loading…
@wang2yn84
[seq packing] move to unreduced loss
#1645 opened Jul 7, 2026 by copybara-service Bot Loading…
@wang2yn84
1
Refactor checkpointing to use Orbax v1.
#1638 opened Jul 1, 2026 by copybara-service Bot Loading…
@tianshub
Verify model supports segment-ids
#1632 opened Jun 30, 2026 by yatlas20098 Collaborator Loading…
6 tasks done
@tianshub
2
Fix beam search test helper indexing
#1630 opened Jun 28, 2026 by fallintoplace Contributor 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
Add cli for frozenlake gemma4-e2b-it recipe
#1626 opened Jun 26, 2026 by lc5211 Collaborator Loading…
6 tasks done
@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
Add chunked prefill with prefix KV cache reuse to Tunix Gemma 4.
#1597 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
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
3
ProTip! Exclude everything labeled bug with -label:bug.