-
Notifications
You must be signed in to change notification settings - Fork 227
Pull requests: google/tunix
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
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…
Add GRPO natural language-to-SQL example with execution-based reward
#997
opened Jan 22, 2026 by
NP2241
Loading…
6 tasks done
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
[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
[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…
[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
[Not ready for review] Memory Optimizations on Gemma2 2B
#966
opened Jan 12, 2026 by
Green0-0
Loading…
fix: Add truncation support to pad_to_length for consistent prompt shapes
#964
opened Jan 11, 2026 by
Cascoopman
Loading…
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
[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…
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…
Previous Next
ProTip!
Mix and match filters to narrow down what you’re looking for.