close
Skip to content

Pull requests: google/tunix

New pull request New
161 Open 1,168 Closed
161 Open 1,168 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Relax batch size constraints when compute ref_logps to consumer
#1431 opened Apr 24, 2026 by copybara-service Bot Loading…
@abheesht17
Fix deleting destination buffers during reshard.
#1430 opened Apr 24, 2026 by NicoGrande Collaborator Loading…
6 tasks done
@abheesht17
Update model name in Gemma2 config for SFT and RL.
#1429 opened Apr 24, 2026 by copybara-service Bot Loading…
@abheesht17
Prevent adding environment tokens for terminal steps in trajectory collection.
#1427 opened Apr 23, 2026 by copybara-service Bot Loading…
@abheesht17
Update training dataset to R2E-Gym-Subset.
#1426 opened Apr 23, 2026 by copybara-service Bot Loading…
@abheesht17
populate deepswe train branch to main
#1420 opened Apr 21, 2026 by copybara-service Bot Loading…
@sizhit2
fix PEFT trainer early exit
#1418 opened Apr 20, 2026 by copybara-service Bot Loading…
@sizhit2
Rename build_and_test_tunix.yml to build_andtest_tunix.yml
#1417 opened Apr 17, 2026 by Tsukimarf Loading…
3 of 6 tasks
@sizhit2
Gemma4 sampling optimizations.
#1415 opened Apr 17, 2026 by copybara-service Bot Loading…
@sizhit2
1
seq_packing_benchmark
#1414 opened Apr 17, 2026 by copybara-service Bot Loading…
@sizhit2
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…
@hgao327
Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#1409 opened Apr 14, 2026 by copybara-service Bot Loading…
@hgao327
Introduce checkpoint_options to manage options for V1 Orbax Checkpointing.
#1408 opened Apr 14, 2026 by copybara-service Bot Loading…
@hgao327
Migrate Tunix notebook examples to Orbax Checkpoint v1 API.
#1407 opened Apr 14, 2026 by copybara-service Bot Loading…
@hgao327
Update DeepScaler and DeepSWE examples to use Orbax v1 checkpointing APIs.
#1404 opened Apr 13, 2026 by copybara-service Bot Loading…
@hgao327
Migrate automodel class to use Orbax Checkpoint V1 API.
#1403 opened Apr 13, 2026 by copybara-service Bot Loading…
@hgao327
Migrate distillation and qlora examples to use Orbax v1 checkpointing APIs.
#1402 opened Apr 13, 2026 by copybara-service Bot Loading…
@hgao327
Migrate Tunix gemma params checkpoint loading to Orbax V1
#1401 opened Apr 13, 2026 by copybara-service Bot Loading…
@hgao327
Update Tunix tests that leverage ocp.CheckpointManagerOptions to instead use tunix.sft.checkpoint_options.
#1396 opened Apr 10, 2026 by copybara-service Bot Loading…
@hgao327
Split TPU tests workflow to 3 parts
#1393 opened Apr 10, 2026 by copybara-service Bot Loading…
@hgao327
fix: use jnp.exp(log_probs) instead of softmax(log_probs) in compute_entropy_from_logits
#1387 opened Apr 10, 2026 by kuishou68 Loading…
1
@hgao327
Update pyproject.toml
#1379 opened Apr 9, 2026 by syrkis Loading…
@hgao327
Add RLOO (REINFORCE Leave-One-Out) learner for lower-variance policy …
#1377 opened Apr 9, 2026 by kbhujbal Loading…
6 tasks done
1
@hgao327
support fsdp packing in sequence packing
#1370 opened Apr 8, 2026 by copybara-service Bot Loading…
@lc5211
DO NOT SUBMIT: testing gemma4
#1364 opened Apr 7, 2026 by copybara-service Bot Loading…
@lc5211
ProTip! What’s not been updated in a month: updated:<2026-03-25.