Skip to content

Pull requests: google/tunix

New pull request New
102 Open 874 Closed
102 Open 874 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

Use the rollout engine construct mesh.
#1060 opened Feb 7, 2026 by wang2yn84 Loading…
6 tasks
1
Refactor tunix Gemma3-4b SFT script to use new config structure.
#1059 opened Feb 7, 2026 by copybara-service bot Loading…
1
Log the computed score in GSM8K reward function
#1058 opened Feb 7, 2026 by copybara-service bot Loading…
1
Add logging for perf metric mode
#1057 opened Feb 6, 2026 by copybara-service bot Loading…
Skip softmax and sorting of probabilities when top_p == 1.0 and top_k is None.
#1056 opened Feb 6, 2026 by copybara-service bot Loading…
Add Agent Sandbox integration example
#1055 opened Feb 6, 2026 by igooch • Draft
6
reduce naming support to HF & config_id
#1053 opened Feb 6, 2026 by copybara-service bot Loading…
[Tunix] Refactor DeepScaler training script to support different rollout engines and mesh configurations.
#1048 opened Feb 5, 2026 by copybara-service bot Loading…
create two perfetto verions
#1043 opened Feb 4, 2026 by copybara-service bot Loading…
[Tiny Feat] add rollout_sglang_jax_log_level in RolloutConfig
#1041 opened Feb 3, 2026 by aolemila Loading…
6 tasks
1
4
feat: log rollout and train time at micro batch level.
#1038 opened Feb 3, 2026 by copybara-service bot Loading…
[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…
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…
update distillation nnx usage
#1023 opened Jan 29, 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
7
Add max_context_tokens to trajectory engine.
#1005 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…
ProTip! Add no:assignee to see everything that’s not assigned.