Skip to content

Pull requests: google/tunix

New pull request New
101 Open 849 Closed
101 Open 849 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

[Tunix] Use compat.ModuleDict for Flax nnx.Dict compatibility.
#1033 opened Jan 31, 2026 by copybara-service bot Loading…
Lazily import reward_manager in function_registry.
#1032 opened Jan 30, 2026 by copybara-service bot Loading…
support customized agent and env impl in agentic_rl_learner
#1031 opened Jan 30, 2026 by copybara-service bot Loading…
ping dep version
#1028 opened Jan 30, 2026 by copybara-service bot Loading…
Add support for stop strings in vLLM sampler and rollout.
#1027 opened Jan 30, 2026 by copybara-service bot Loading…
Add Gemini Code Assist style guide for PR reviews
#1024 opened Jan 29, 2026 by copybara-service bot Loading…
6
update distillation nnx usage
#1023 opened Jan 29, 2026 by copybara-service bot Loading…
fix model_call in agentic_rl_learner
#1021 opened Jan 29, 2026 by copybara-service bot Loading…
[Tunix] Fix Qwen3 vLLM to JAX lm_head mapping.
#1020 opened Jan 29, 2026 by copybara-service bot Loading…
move scrubber after transformation for it to locate code in github
#1016 opened Jan 28, 2026 by copybara-service bot Loading…
fix transformations with docs
#1015 opened Jan 28, 2026 by copybara-service bot Loading…
[Tunix] Support scanned to unscanned weight transfer in transfer_state_directly
#1008 opened Jan 28, 2026 by gagika • Draft
6 tasks done
Add max_context_tokens to trajectory engine.
#1005 opened Jan 27, 2026 by copybara-service bot Loading…
Implement VLLM JAX mappings for Qwen3 models to allow weight updates
#1004 opened Jan 27, 2026 by copybara-service bot Loading…
Allow config_id as an alternative model_id to automodel
#1002 opened Jan 27, 2026 by copybara-service bot Loading…
fix(rl): robust integer validation for utils.py (Fixes #953)
#1000 opened Jan 24, 2026 by abdulwahabahmedkhanyusufzai Loading…
2
2
Merge duplicate example content
#999 opened Jan 23, 2026 by copybara-service bot Loading…
Add GRPO natural language-to-SQL example with execution-based reward
#997 opened Jan 22, 2026 by NP2241 Loading…
6 tasks done
1
remove pip install jax==0.8.1 flax==0.12.0 libtpu==0.0.24
#996 opened Jan 22, 2026 by aolemila Loading…
6 tasks
1
doc: fix indentation in README.md.
#993 opened Jan 21, 2026 by copybara-service bot Loading…
[Tunix] Improve the reshard logging. Current logging only polls on the results for on device finish time. We need to do the same thing for the inputs for accurate logging. This cl splits the total wait time to waiting for input time and output ready time.
#992 opened Jan 21, 2026 by copybara-service bot Loading…
[WIP] VLM Training (1): Add vision to Gemma 3
#986 opened Jan 18, 2026 by copybara-service bot Loading…
[Tunix] Add trajectory data logging to disk for further analysis and visualization.
#980 opened Jan 15, 2026 by copybara-service bot Loading…
Feature/gdpo
#978 opened Jan 15, 2026 by kmsgnnew Loading…
6 tasks done
1
[Tunix] Fix the safetensors saving. When loading the base model state, it should load all the safetensor file instead of a hardcoded one.
#972 opened Jan 13, 2026 by copybara-service bot Loading…
ProTip! What’s not been updated in a month: updated:<2026-01-01.