Skip to content

Pull requests: google/tunix

New pull request New
77 Open 804 Closed
77 Open 804 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

Fix the cli to have the proper model config over written.
#960 opened Jan 10, 2026 by wang2yn84 Loading…
6 tasks done
4
[internal] extend XM launch to peft
#959 opened Jan 9, 2026 by copybara-service bot Loading…
Fix #802: Add google-cloud-storage and gcsfs to dependencies
#958 opened Jan 9, 2026 by Aaryan-549 Loading…
3 of 5 tasks
1
adding missing test case variant
#956 opened Jan 8, 2026 by copybara-service bot Loading…
Add overlong reward shaping for DAPO.
#947 opened Jan 6, 2026 by copybara-service bot Loading…
Add CRD distillation strategy and tests
#933 opened Dec 22, 2025 by surajyadav-research Loading…
1
[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
[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…
[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…
5
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
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
Feat/add lora for sglangjax
#826 opened Dec 2, 2025 by aolemila Loading…
6 tasks
1
@aolemila
19
ProTip! Follow long discussions with comments:>50.