Pull requests: google/tunix
Pull requests list
Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#1366
opened Apr 7, 2026 by
copybara-service
bot
Loading…
Add skip_sharding_optimizer flag to allow passing in a pre-sharded optimizer
#1361
opened Apr 6, 2026 by
igorts-git
•
Draft
6 tasks
[Tunix] Improve learning rate extraction in PEFT trainer.
#1353
opened Apr 2, 2026 by
copybara-service
bot
Loading…
add knobs for perf metrics in
grpo_demo_llama3_qwen2.py
#1348
opened Apr 2, 2026 by
copybara-service
bot
Loading…
Move RLOO advantage estimator from GRPOLearner to AgenticGRPOLearner.
#1340
opened Apr 1, 2026 by
copybara-service
bot
Loading…
Fix LoRA merge for sharded Gemma 3 checkpoints and HF parameter mapping
#1336
opened Apr 1, 2026 by
ayehninnkhine
Loading…
Simplify tunix/oss file transformations in copy.bara.sky.
#1335
opened Apr 1, 2026 by
copybara-service
bot
Loading…
Introduce a checkpoint_options compatibility layer to manage options for V1 Orbax Checkpointing.
#1330
opened Mar 31, 2026 by
copybara-service
bot
Loading…
[tunix/sft] Add
shard_input_data flag to PeftTrainer
#1325
opened Mar 28, 2026 by
copybara-service
bot
Loading…
Fix Qwen2 KV-cache dtype mismatch in cached decoding
#1322
opened Mar 27, 2026 by
skwh54
Loading…
6 tasks done
improve rl_cluster init corner case handling and test coverage
#1311
opened Mar 26, 2026 by
copybara-service
bot
Loading…
fix file path for window, : is reserved name on window
#1307
opened Mar 26, 2026 by
HugoTian
Loading…
security: replace eval() with safe AST math evaluator in calculate reward
#1279
opened Mar 23, 2026 by
RaymondSeven
Loading…
ProTip!
Updated in the last three days: updated:>2026-04-04.