Skip to content

Pull requests: google/tunix

New pull request New
238 Open 1,530 Closed
238 Open 1,530 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

[Draft] Add MLPerf RCP logging for post training in DeepSWE
#1883 opened Aug 13, 2026 by susanbao Collaborator • Draft
@tianshub
Enhance Tunix distributed RL orchestrator for agentic workflows
#1882 opened Aug 13, 2026 by copybara-service Bot Loading…
@tianshub
[tunix] Preserve optimizer-state dtypes across updates
#1880 opened Aug 13, 2026 by ethannnnnn Contributor Loading…
@tianshub
Add Tunix SFT fine-tuning pipeline and ROC conversion tooling for Gemma 4 models.
#1879 opened Aug 12, 2026 by copybara-service Bot Loading…
@jiangyangmu
1
Refactor Tunix Gemma 4 model into focused modules with backward-compatible facade.
#1878 opened Aug 12, 2026 by copybara-service Bot Loading…
@jiangyangmu
1
Enforce zero-ignore static typing rigor across ShardingConfig and support activation checkpointing in Tunix Gemma 4.
#1877 opened Aug 12, 2026 by copybara-service Bot Loading…
@jiangyangmu
1
Fix missing MaxText dependencies in TPU nightly regression workflow.
#1876 opened Aug 11, 2026 by copybara-service Bot Loading…
@jiangyangmu
Introduce ragged paged KV cache.
#1875 opened Aug 11, 2026 by copybara-service Bot Loading…
@jiangyangmu
1
feat: add mcjax YAML template to distributed module.
#1874 opened Aug 11, 2026 by copybara-service Bot Loading…
@jiangyangmu
Make the rollout worker a weight sync destination, with a fail-closed round tracker
#1872 opened Aug 10, 2026 by copybara-service Bot Loading…
@jiangyangmu
1
Fix raiden visibility to support tunix integration
#1869 opened Aug 10, 2026 by copybara-service Bot Loading…
@jiangyangmu
1
**DO NOT SUBMIT** SIGILL in WightSynchronizerBase
#1866 opened Aug 10, 2026 by copybara-service Bot Loading…
@jiangyangmu
1
feat: add persistent connection API to discovery service.
#1863 opened Aug 8, 2026 by copybara-service Bot Loading…
@jiangyangmu
Extract configs into tunix/common/configs.py
#1861 opened Aug 8, 2026 by copybara-service Bot Loading…
@jiangyangmu
Extract configs into tunix/common/configs.py
#1859 opened Aug 8, 2026 by copybara-service Bot Loading…
@jiangyangmu
Add ATIF Trajectory converter
#1857 opened Aug 7, 2026 by copybara-service Bot Loading…
@jiangyangmu
1
[Tunix] Add support for leave-one-out baseline in GRPO advantage estimation.
#1856 opened Aug 7, 2026 by copybara-service Bot Loading…
@jiangyangmu
Update vLLM TPU container image to 0.26.0 in Tunix CI workflows and add JAX sharding shims
#1855 opened Aug 7, 2026 by copybara-service Bot Loading…
@jiangyangmu
Adding orchestrator basic implementation to examples
#1853 opened Aug 7, 2026 by copybara-service Bot Loading…
@jiangyangmu
Align optimizer state dtypes in PEFT trainer.
#1844 opened Aug 6, 2026 by copybara-service Bot Loading…
Add Frozenlake tests to TPU nightly regression workflow.
#1843 opened Aug 6, 2026 by copybara-service Bot Loading…
RLDriver and DistributedRLEngine
#1841 opened Aug 6, 2026 by copybara-service Bot Loading…
@jiangyangmu
[Tunix] Add Frozenlake tests to TPU nightly regression and update dependencies.
#1831 opened Aug 5, 2026 by copybara-service Bot Loading…
@abheesht17
[Tunix] Move Dockerfile and requirements install script changes.
#1828 opened Aug 4, 2026 by copybara-service Bot Loading…
@abheesht17
Add vLLM JAX mapping for Gemma models.
#1826 opened Aug 4, 2026 by copybara-service Bot Loading…
@abheesht17
ProTip! no:milestone will show everything without a milestone.