Skip to content

Pull requests: google/tunix

New pull request New
217 Open 1,420 Closed
217 Open 1,420 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

fix(sft): eliminate GradientAccumulator compile-time regression
#1751 opened Jul 24, 2026 by copybara-service Bot Loading…
@sizhit2
1
Add OpenTelemetry (OTel) endpoint injection to Tunix deployment pipeline.
#1750 opened Jul 23, 2026 by copybara-service Bot Loading…
@sizhit2
[tunix] Add exact prepared diffusion policy optimization
#1749 opened Jul 23, 2026 by ethannnnnn Loading…
@sizhit2
1
[tunix] Add prepared diffusion policy scoring
#1748 opened Jul 23, 2026 by ethannnnnn Loading…
@sizhit2
1
[tunix] Add weighted prepared diffusion OPD objective
#1747 opened Jul 23, 2026 by ethannnnnn Loading…
@sizhit2
1
[tunix] Add prepared diffusion distillation batches
#1746 opened Jul 23, 2026 by ethannnnnn Loading…
@sizhit2
1
[tunix] Add target-aligned diffusion SFT adapter
#1745 opened Jul 23, 2026 by ethannnnnn Loading…
@sizhit2
1
[tunix] Accumulate weighted loss gradients globally
#1744 opened Jul 23, 2026 by ethannnnnn Loading…
@sizhit2
1
[tunix] Add model-agnostic JAX diffusion training contracts
#1743 opened Jul 23, 2026 by ethannnnnn Loading…
@sizhit2
1
Fix dependency errors and deprecated JAX syntax in qlora_gemma notebook
#1740 opened Jul 23, 2026 by selamw1 Collaborator Loading…
@sizhit2
Fix vl_dpo_gemma3 notebook execution bugs, JAX deprecations, and OOM memory allocation
#1739 opened Jul 23, 2026 by selamw1 Collaborator Loading…
@sizhit2
test: add unit tests to distributed modules
#1736 opened Jul 23, 2026 by copybara-service Bot Loading…
@sizhit2
Fix FrozenLake shuffle_data parsing
#1734 opened Jul 23, 2026 by kiwigitops Loading…
@sizhit2
Update tpu-inference and vllm pinned commits.
#1723 opened Jul 22, 2026 by copybara-service Bot Loading…
@hgao327
Fix AttributeError with the SentencePieceProcessor during tokenizer initialization
#1719 opened Jul 21, 2026 by rajasekharporeddy Collaborator Loading…
4 tasks done
@hgao327
Internal
#1716 opened Jul 20, 2026 by copybara-service Bot Loading…
@hgao327
Internal
#1712 opened Jul 20, 2026 by copybara-service Bot Loading…
@hgao327
Add TrajectoryQueueManager with pluggable grouping and filtering.
#1708 opened Jul 17, 2026 by copybara-service Bot Loading…
@hgao327
[WIP] Weight Sync
#1704 opened Jul 17, 2026 by YixuanWang-99 Collaborator Loading…
6 tasks
@hgao327
Fix docker build and add training dependencies (kubernetes, r2e-gym)
#1703 opened Jul 17, 2026 by niting Contributor Loading…
@hgao327
2
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] Add opt-in OpenTelemetry double-write to MetricsLogger
#1686 opened Jul 15, 2026 by erain Loading…
6 tasks done
@sizhit2 @lc5211
6
Yatlas support all models ppo
#1678 opened Jul 14, 2026 by yatlas20098 Collaborator Loading…
6 tasks done
@lc5211
support internal gemma4 ckpt in config
#1674 opened Jul 14, 2026 by copybara-service Bot Loading…
@lc5211
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.