Skip to content

Pull requests: google/tunix

New pull request New
89 Open 832 Closed
89 Open 832 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

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
1
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…
[Distillation] Enhanced Logit Strategy with Top-K and Metrics
#991 opened Jan 21, 2026 by gagika Loading…
6 tasks done
@tianshub @wang2yn84
[WIP] VLM Training (1): Add vision to Gemma 3
#986 opened Jan 18, 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…
ProTip! Mix and match filters to narrow down what you’re looking for.