Skip to content

Pull requests: google/tunix

New pull request New
180 Open 1,283 Closed
180 Open 1,283 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

Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer and. migrate distill_utils to v1 orbax semantics.
#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
Add Qwen3-1.7B GSM8K VTC GRPO demo recipe.
#1569 opened Jun 8, 2026 by copybara-service Bot Loading…
@sizhit2
Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#1567 opened Jun 5, 2026 by copybara-service Bot Loading…
@sizhit2
Undo transformers<=4.57.1 pin
#1566 opened Jun 5, 2026 by ringohoffman Loading…
6 tasks done
@sizhit2
Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#1565 opened Jun 5, 2026 by copybara-service Bot Loading…
@sizhit2
[Tunix] A few fixes to Gemma4 model implementation.
#1554 opened Jun 1, 2026 by copybara-service Bot Loading…
@hgao327
fix(rl): accept numpy ints and reject bool in is_positive_integer
#1552 opened May 29, 2026 by kitrakrev Loading…
2 tasks done
@hgao327
1
[Tunix] Introduced the custom Gemma4ChatTemplateParser which parses the multi-turn messages correctly.
#1551 opened May 29, 2026 by copybara-service Bot Loading…
@hgao327
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
2
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
Add RL smoke tests to Tunix TPU CI pipeline
#1516 opened May 14, 2026 by copybara-service Bot Loading…
@wang2yn84
[seq packing] move to unreduced loss
#1514 opened May 13, 2026 by copybara-service Bot Loading…
@tianshub
Add Pathways support and proper testing.
#1513 opened May 13, 2026 by copybara-service Bot Loading…
@tianshub
Refactor tests to common base file to support additional pathways testing.
#1512 opened May 13, 2026 by copybara-service Bot Loading…
@tianshub
add support for gemma4 maxtext path.
#1506 opened May 12, 2026 by NicoGrande Collaborator • Draft
6 tasks
@tianshub
[Seq Packing] stream based gradient accumulation
#1505 opened May 12, 2026 by copybara-service Bot Loading…
@tianshub
move to unreduced loss fn
#1502 opened May 8, 2026 by copybara-service Bot Loading…
@tianshub
add qwen3-235b config
#1484 opened May 5, 2026 by copybara-service Bot Loading…
@jiangyangmu
ProTip! Filter pull requests by the default branch with base:main.