Skip to content

Pull requests: google/tunix

New pull request New
160 Open 1,155 Closed
160 Open 1,155 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

Rename build_and_test_tunix.yml to build_andtest_tunix.yml
#1417 opened Apr 17, 2026 by Tsukimarf Loading…
3 of 6 tasks
@sizhit2
Fix agentic cli dataloading
#1416 opened Apr 17, 2026 by wang2yn84 Collaborator Loading…
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
Internal
#1390 opened Apr 10, 2026 by copybara-service bot Loading…
@hgao327
1
check if limit reached more proactivately
#1388 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
Adding support for fused moe kernels in sampler.
#1385 opened Apr 9, 2026 by NicoGrande Collaborator • Draft
6 tasks
@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
[Tunix] Improve learning rate extraction in PEFT trainer.
#1353 opened Apr 2, 2026 by copybara-service bot Loading…
@lc5211
Openmath fixes
#1341 opened Apr 1, 2026 by xianglon-commits Loading…
6 tasks
@wang2yn84
transpose flash attention projection back
#1338 opened Apr 1, 2026 by copybara-service bot Loading…
@wang2yn84
ProTip! Add no:assignee to see everything that’s not assigned.