Skip to content

Pull requests: google/tunix

New pull request New
78 Open 786 Closed
78 Open 786 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

enable proper report for learning rate during training
#940 opened Dec 26, 2025 by Hanjun-Dai Loading…
6 tasks done
Fix broken example script references in CLI README
#938 opened Dec 23, 2025 by SarveshMahalingam Loading…
5 of 6 tasks
1
add llama 3.2 1b/3b-instruct model config
#935 opened Dec 23, 2025 by Hanjun-Dai Loading…
6 tasks done
4
Add CRD distillation strategy and tests
#933 opened Dec 22, 2025 by surajyadav-research Loading…
[Tunix] Update dataset initialization in nightly regression script and add to CI.
#932 opened Dec 22, 2025 by copybara-service bot Loading…
fix(utils): correct positive integer validation
#930 opened Dec 21, 2025 by parthmaheshwari-hub Loading…
1
Refactor tokenizer adapter to support special EOS token
#927 opened Dec 21, 2025 by kamalkraj Loading…
6 tasks done
refactor rl config: 1) remove unnecessary params; 2) model_config consistency
#926 opened Dec 20, 2025 by Hanjun-Dai Loading…
6 tasks done
4
[Tunix] Remove intermediate activation sharding in FeedForward blocks.
#921 opened Dec 18, 2025 by copybara-service bot Loading…
[Tunix] Add agentic mode support to grpo_demo_llama3_qwen2.py.
#920 opened Dec 18, 2025 by copybara-service bot Loading…
remove mesh cm from example
#908 opened Dec 17, 2025 by copybara-service bot Loading…
[Tunix] Reorder TPU nightly regression workflow steps.
#906 opened Dec 17, 2025 by copybara-service bot Loading…
Fix AttributeError in is_positive_integer on Python 3.11
#903 opened Dec 17, 2025 by bug39 Loading…
4
fix broken test.
#898 opened Dec 16, 2025 by copybara-service bot Loading…
Upgrade GitHub Actions to latest versions
#895 opened Dec 16, 2025 by salmanmkc Loading…
4
Upgrade GitHub Actions for Node 24 compatibility
#894 opened Dec 16, 2025 by salmanmkc Loading…
7
Fdpo
#892 opened Dec 16, 2025 by xinzhuo20 Loading…
3
Add metrics support to CheckpointManager and PeftTrainer
#885 opened Dec 14, 2025 by kamalkraj Loading…
6 tasks done
10
Add Qwen3 235B-A22B model config
#871 opened Dec 10, 2025 by yhtang Loading…
6 tasks done
1
fix sft smoke test
#854 opened Dec 8, 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…
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
10
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
ProTip! Follow long discussions with comments:>50.