Skip to content

Pull requests: google/tunix

New pull request New
83 Open 824 Closed
83 Open 824 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

[WIP] VLM Training (1): Add vision to Gemma 3
#986 opened Jan 18, 2026 by copybara-service bot Loading…
Disable Lora in base_config.yaml
#983 opened Jan 15, 2026 by copybara-service bot Loading…
Unify to HF as the single model_id
#981 opened Jan 15, 2026 by copybara-service bot Loading…
[Tunix] Add trajectory data logging to disk for further analysis and visualization.
#980 opened Jan 15, 2026 by copybara-service bot Loading…
Feature/gdpo
#978 opened Jan 15, 2026 by kmsgnnew Loading…
6 tasks done
1
[Tunix] Fix the safetensors saving. When loading the base model state, it should load all the safetensor file instead of a hardcoded one.
#972 opened Jan 13, 2026 by copybara-service bot Loading…
Fix PeftTrainer input sharding for raw inputs and add regression test
#967 opened Jan 12, 2026 by jayshrivastava0 Loading…
6 tasks done
1
[Not ready for review] Memory Optimizations on Gemma2 2B
#966 opened Jan 12, 2026 by Green0-0 Loading…
1
fix: Add truncation support to pad_to_length for consistent prompt shapes
#964 opened Jan 11, 2026 by Cascoopman Loading…
Fix batch_size=1 token_buffer truncation in sampler
#963 opened Jan 11, 2026 by Cascoopman Loading…
1
feat(agentic): Add tool calling support for GRPO training
#961 opened Jan 11, 2026 by Cascoopman 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
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
ProTip! Filter pull requests by the default branch with base:main.