Pull requests: google/tunix
Pull requests list
Allow various length distributions in mock rollout
#1398
opened Apr 10, 2026 by
copybara-service
bot
Loading…
Update Checkpointing Options construction through CLI configuration to move from Orbax V0 CheckpointManagerOptions to Tunix's CheckpointOptions.
#1397
opened Apr 10, 2026 by
copybara-service
bot
Loading…
Update Tunix tests that leverage ocp.CheckpointManagerOptions to instead use tunix.sft.checkpoint_options.
#1396
opened Apr 10, 2026 by
copybara-service
bot
Loading…
Remove the
step_time parameter from the SFT TrainingHooks.on_train_step_end function.
#1395
opened Apr 10, 2026 by
copybara-service
bot
Loading…
Split TPU tests workflow to 3 parts and presubmit bug fix
#1393
opened Apr 10, 2026 by
copybara-service
bot
Loading…
Check if GCS path and raise error in v1 perf tracer
#1389
opened Apr 10, 2026 by
copybara-service
bot
Loading…
check if limit reached more proactivately
#1388
opened Apr 10, 2026 by
copybara-service
bot
Loading…
fix: use jnp.exp(log_probs) instead of softmax(log_probs) in compute_entropy_from_logits
#1387
opened Apr 10, 2026 by
kuishou68
Loading…
Adding support for fused moe kernels in sampler.
#1385
opened Apr 9, 2026 by
NicoGrande
Collaborator
•
Draft
6 tasks
Add RLOO (REINFORCE Leave-One-Out) learner for lower-variance policy …
#1377
opened Apr 9, 2026 by
kbhujbal
Loading…
6 tasks done
Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#1366
opened Apr 7, 2026 by
copybara-service
bot
Loading…
[Tunix] Improve learning rate extraction in PEFT trainer.
#1353
opened Apr 2, 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…
ProTip!
no:milestone will show everything without a milestone.