Pull requests: google/tunix
Pull requests list
Include vLLM and tpu inference to Tunix build.
#1411
opened Apr 15, 2026 by
wang2yn84
Collaborator
Loading…
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…
Add DeepSWE evaluation script with ActionGuard.
#1406
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…
Allow various length distributions in mock rollout
#1398
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…
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
[Tunix] Improve learning rate extraction in PEFT trainer.
#1353
opened Apr 2, 2026 by
copybara-service
bot
Loading…
ProTip!
Filter pull requests by the default branch with base:main.