Skip to content

Pull requests: google/tunix

New pull request New
211 Open 1,373 Closed
211 Open 1,373 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

GRPO: pluggable per-token log-prob hook (for non-autoregressive / diffusion rollouts)
#1698 opened Jul 17, 2026 by ethannnnnn Loading…
@hgao327
1
[DO NOT SUBMIT]
#1697 opened Jul 17, 2026 by copybara-service Bot Loading…
@hgao327
[Tunix] Orchestrator: Introducing common datatypes
#1696 opened Jul 17, 2026 by copybara-service Bot Loading…
@hgao327
[Tunix] Restrict JAX version in tunix/oss pyproject.toml.
#1695 opened Jul 16, 2026 by copybara-service Bot Loading…
@hgao327
Migrate Tunix's SFT Checkpoint Manager usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#1690 opened Jul 15, 2026 by copybara-service Bot Loading…
@lc5211
prototype only do not submit
#1689 opened Jul 15, 2026 by copybara-service Bot Loading…
@lc5211
[Tunix] Adopt OpenTelemetry behind MetricsLogger
#1686 opened Jul 15, 2026 by erain Loading…
6 tasks done
1
@lc5211
Enable PPO for all model types and add PPO examples.
#1683 opened Jul 15, 2026 by copybara-service Bot Loading…
@lc5211
1
Yatlas support all models ppo
#1678 opened Jul 14, 2026 by yatlas20098 Collaborator Loading…
6 tasks done
@lc5211
Add raw completion length metrics to Agentic GRPO Learner
#1675 opened Jul 14, 2026 by copybara-service Bot Loading…
@lc5211
support internal gemma4 ckpt in config
#1674 opened Jul 14, 2026 by copybara-service Bot Loading…
@lc5211
[Seq Packing] Unified sequence packing: packed logprob + pack-first (agentic GRPO)
#1673 opened Jul 14, 2026 by copybara-service Bot Loading…
@lc5211
1
[Seq Packing] KK load balancing: balance packs across DP shards (P5)
#1672 opened Jul 14, 2026 by copybara-service Bot Loading…
@lc5211
1
[Seq Packing] Loss normalization under packing (P3): unpack aggregation + global grad weighting
#1671 opened Jul 14, 2026 by copybara-service Bot Loading…
@lc5211
1
[Tuinx] garbage collect after update_param, for both offloading case and weight sync.
#1670 opened Jul 13, 2026 by copybara-service Bot Loading…
@lc5211
Internal
#1668 opened Jul 13, 2026 by copybara-service Bot Loading…
@lc5211
Update out-of-tokens prompt for SWE agent.
#1666 opened Jul 13, 2026 by copybara-service Bot Loading…
@lc5211
[Seq Packing] KK load balancing: balance packs across DP shards (P5)
#1665 opened Jul 13, 2026 by yuxuandexter Collaborator Loading…
@lc5211
Internal
#1664 opened Jul 13, 2026 by copybara-service Bot Loading…
@lc5211
1
[Seq Packing] Loss normalization under packing (P3): unpack aggregation + global grad weighting
#1662 opened Jul 10, 2026 by yuxuandexter Collaborator Loading…
@lc5211
[Seq Packing] Unified sequence packing: packed logprob + pack-first (agentic GRPO)
#1659 opened Jul 10, 2026 by yuxuandexter Collaborator Loading…
@lc5211
1
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
ProTip! Adding no:label will show everything without a label.