-
Notifications
You must be signed in to change notification settings - Fork 258
Pull requests: google/tunix
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
[Tunix] Update math eval script with vLLM sampler configuration to use vllm_jax backend and engine_kwargs. Adjust seeding for vLLm.
#1246
opened Mar 13, 2026 by
copybara-service
bot
Loading…
[Tunix] Implement batch logging and add signal handlers for graceful shutdown for trajectory logger.
#1245
opened Mar 13, 2026 by
copybara-service
bot
Loading…
use timeline snapshot for export to avoid race condition
#1240
opened Mar 12, 2026 by
copybara-service
bot
Loading…
allow passing eval_ds parameters through tunix cli, plus bugfixes
#1238
opened Mar 12, 2026 by
andytwigg
Loading…
Update put_params_on_memory_kind in tunix/rl/utils.py
#1235
opened Mar 11, 2026 by
copybara-service
bot
Loading…
Add sglang vllm config and update rollout engine setup in train_deepswe_nb.py.
#1222
opened Mar 10, 2026 by
copybara-service
bot
Loading…
Add retries to Hugging Face Hub calls in tunix/oss/utils.py to mitigate gateway errors.
#1213
opened Mar 7, 2026 by
copybara-service
bot
Loading…
Colab now has dotenv by default so this check is wrong
#1202
opened Mar 6, 2026 by
copybara-service
bot
Loading…
[Tunix] add an enable_hlo_proto option to ProfilerOptions to control whether HLO protos are included in the profile, potentially reducing profile size.
#1201
opened Mar 6, 2026 by
copybara-service
bot
Loading…
Harden calculate_reward: replace unsafe eval() with safe arithmetic evaluator
#1197
opened Mar 5, 2026 by
0xmrma
Loading…
Extending scanned to unscanned support for GPT-OSS like models.
#1189
opened Mar 3, 2026 by
NicoGrande
Loading…
6 tasks done
Fix sharding rank mismatch for LoRA weights and optimizer states in
qlora_gemma.ipynb notebook
#1186
opened Mar 3, 2026 by
rajasekharporeddy
Loading…
4 tasks done
Fix sharding rank mismatch for LoRA weights and optimizer states in
grpo_gemma.ipynb notebook
#1180
opened Mar 2, 2026 by
rajasekharporeddy
Loading…
4 tasks done
fix(rl): flush final partial micro-batch to avoid sample drop
#1179
opened Mar 1, 2026 by
pjo256
Loading…
6 tasks done
[Tunix] Clear and reinitialize vLLM KV cache when loading weights.
#1167
opened Feb 26, 2026 by
copybara-service
bot
Loading…
fix: add thread safety locks to perf tracing module
#1166
opened Feb 26, 2026 by
kbhujbal
Loading…
6 tasks
Previous Next
ProTip!
no:milestone will show everything without a milestone.