Skip to content

Pull requests: google/tunix

New pull request New
103 Open 853 Closed
103 Open 853 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

[Tiny Feat] add rollout_sglang_jax_log_level in RolloutConfig
#1041 opened Feb 3, 2026 by aolemila Loading…
6 tasks
1
4
feat: add multi-rollout engine interfaces.
#1039 opened Feb 3, 2026 by copybara-service bot Loading…
feat: log rollout and train time at micro batch level.
#1038 opened Feb 3, 2026 by copybara-service bot Loading…
bug: fix metric refer_inference_time.
#1037 opened Feb 3, 2026 by copybara-service bot Loading…
Add Tunix to perfetto visibility list
#1036 opened Feb 3, 2026 by copybara-service bot Loading…
Collect event data from PerfMetrics
#1035 opened Feb 2, 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…
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…
update distillation nnx usage
#1023 opened Jan 29, 2026 by copybara-service bot Loading…
[Tunix] Fix Qwen2/3 vLLM to JAX lm_head mapping.
#1020 opened Jan 29, 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…
VLM Training (1): Add vision to Gemma 3
#986 opened Jan 18, 2026 by copybara-service bot Loading…
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.