Skip to content

Pull requests: google/tunix

New pull request New
195 Open 1,334 Closed
195 Open 1,334 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

Improve the COMPACT mode in mesh creation.
#1641 opened Jul 1, 2026 by wang2yn84 Collaborator Loading…
6 tasks
@tianshub
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
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
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
[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
2
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
ProTip! Adding no:label will show everything without a label.