Skip to content

Pull requests: google/tunix

New pull request New
67 Open 717 Closed
67 Open 717 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] Update parallel sizes to use ROLLOUT_MESH in grpo_demo.
#851 opened Dec 5, 2025 by copybara-service bot Loading…
update gemma2 run script to use the config yaml
#850 opened Dec 5, 2025 by copybara-service bot Loading…
Update sharding for activations to include 'dp' axis. Validated on internal notebook, after enabling (2, 1, 4) for (dp, fsdp, tp), do see weights replicated on dp axis.
#849 opened Dec 5, 2025 by copybara-service bot Loading…
Add XM launch for tunix cli
#848 opened Dec 5, 2025 by copybara-service bot Loading…
[Bug] replace device_put with reshard_tree to keep compatible with Pathways
#845 opened Dec 5, 2025 by aolemila Loading…
6 tasks
1
Support custom MaxText (vLLM) models in sampler and rollout.
#841 opened Dec 4, 2025 by NicoGrande Loading…
6 tasks done
Explicitly cast input_ids to jnp.int32
#838 opened Dec 4, 2025 by copybara-service bot Loading…
Enable multi-process/multi-controller support for GPUs
#834 opened Dec 3, 2025 by yhtang • Draft
4 of 6 tasks
1
5
Add parity test for Agentic GRPO vs GRPO.
#833 opened Dec 3, 2025 by copybara-service bot Loading…
Remove excessive logs from OSS notebooks.
#830 opened Dec 3, 2025 by wang2yn84 Loading…
6 tasks done
Add AgenticRLLearner base class.
#829 opened Dec 3, 2025 by copybara-service bot Loading…
[Feature] Feat/add lora for sglangjax
#826 opened Dec 2, 2025 by aolemila Loading…
6 tasks
1
Support custom MaxText (vLLM) models in sampler and rollout.
#821 opened Dec 1, 2025 by gagika • Draft
6 tasks
Add GRPO micro-batch size validation to prevent incorrect grouping and advantage computation
#820 opened Dec 1, 2025 by aaghaazkhan Loading…
6 tasks done
Add support for interleaved layer mappings and enhanced key mapping regex.
#815 opened Dec 1, 2025 by copybara-service bot Loading…
Add fixed basic inference example using Sampler with dummy Gemma model
#812 opened Nov 29, 2025 by Prasanth-ADS Loading…
1
add automodel to Tunix
#801 opened Nov 25, 2025 by copybara-service bot Loading…
1
Minor Doc Fixes: Correct a typo, add a hyperlink and sequential alignment of contribution steps in documentation.
#800 opened Nov 25, 2025 by rajasekharporeddy Loading…
Support instruction tuned gemma3
#793 opened Nov 25, 2025 by copybara-service bot Loading…
[DNS] debug failure
#781 opened Nov 21, 2025 by copybara-service bot Loading…
Add timing logs to transfer_state.
#764 opened Nov 18, 2025 by copybara-service bot Loading…
Only load from GCS for GCS paths
#756 opened Nov 17, 2025 by copybara-service bot Loading…
Colab & Kaggle Badges
#748 opened Nov 16, 2025 by msghik Loading…
1
Rename build_and_test_tunix.yml to build_andtest_tunix.yml
#747 opened Nov 15, 2025 by Tsukimarf Loading…
3
Speed up tpu-tests
#729 opened Nov 13, 2025 by copybara-service bot Loading…
ProTip! no:milestone will show everything without a milestone.