Pull requests: google/tunix
Pull requests list
Rename build_and_test_tunix.yml to build_andtest_tunix.yml
#1417
opened Apr 17, 2026 by
Tsukimarf
Loading…
3 of 6 tasks
Update Checkpointing Options construction through CLI configuration to move from Orbax V0 CheckpointManagerOptions to Tunix's CheckpointOptions.
#1410
opened Apr 14, 2026 by
copybara-service
Bot
Loading…
Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#1409
opened Apr 14, 2026 by
copybara-service
Bot
Loading…
Introduce checkpoint_options to manage options for V1 Orbax Checkpointing.
#1408
opened Apr 14, 2026 by
copybara-service
Bot
Loading…
Migrate Tunix notebook examples to Orbax Checkpoint v1 API.
#1407
opened Apr 14, 2026 by
copybara-service
Bot
Loading…
Update DeepScaler and DeepSWE examples to use Orbax v1 checkpointing APIs.
#1404
opened Apr 13, 2026 by
copybara-service
Bot
Loading…
Migrate automodel class to use Orbax Checkpoint V1 API.
#1403
opened Apr 13, 2026 by
copybara-service
Bot
Loading…
Migrate distillation and qlora examples to use Orbax v1 checkpointing APIs.
#1402
opened Apr 13, 2026 by
copybara-service
Bot
Loading…
Migrate Tunix gemma params checkpoint loading to Orbax V1
#1401
opened Apr 13, 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…
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
[Tunix] Improve learning rate extraction in PEFT trainer.
#1353
opened Apr 2, 2026 by
copybara-service
Bot
Loading…
ProTip!
What’s not been updated in a month: updated:<2026-03-22.