Pull requests: google/tunix
Pull requests list
Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#1565
opened Jun 5, 2026 by
copybara-service
Bot
Loading…
Fixes the tfds prompt not sanitizated issue for agentic path.
#1556
opened Jun 1, 2026 by
wang2yn84
Collaborator
Loading…
6 tasks
[Tunix] A few fixes to Gemma4 model implementation.
#1554
opened Jun 1, 2026 by
copybara-service
Bot
Loading…
Add Gemma 4 Orbax checkpoint loading support to Tunix.
#1553
opened Jun 1, 2026 by
copybara-service
Bot
Loading…
fix(rl): accept numpy ints and reject bool in is_positive_integer
#1552
opened May 29, 2026 by
kitrakrev
Loading…
2 tasks done
[Tunix] Introduced the custom Gemma4ChatTemplateParser which parses the multi-turn messages correctly.
#1551
opened May 29, 2026 by
copybara-service
Bot
Loading…
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…
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…
shard perfetto trace writes and bound Timeline memory
#1529
opened May 18, 2026 by
copybara-service
Bot
Loading…
Inject unified Tunix and vLLM environment fingerprinting metadata
#1517
opened May 14, 2026 by
copybara-service
Bot
Loading…
Add RL smoke tests to Tunix TPU CI pipeline
#1516
opened May 14, 2026 by
copybara-service
Bot
Loading…
Refactor tests to common base file to support additional pathways testing.
#1512
opened May 13, 2026 by
copybara-service
Bot
Loading…
add support for gemma4 maxtext path.
#1506
opened May 12, 2026 by
NicoGrande
Collaborator
•
Draft
6 tasks
[Seq Packing] stream based gradient accumulation
#1505
opened May 12, 2026 by
copybara-service
Bot
Loading…
fix: Remove interactive login prompts for robust automated runs
#1467
opened May 2, 2026 by
sizhit2
Collaborator
Loading…
ProTip!
What’s not been updated in a month: updated:<2026-05-05.