Skip to content

Pull requests: google/tunix

New pull request New
122 Open 939 Closed
122 Open 939 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

[Resolved 1149] fix oom due to missing closing loop
#1151 opened Feb 25, 2026 by aolemila Loading…
6 tasks
1
@hgao327
1
Auto assign only one reviewer to the pull requests
#1150 opened Feb 25, 2026 by rajasekharporeddy Loading…
4 tasks done
1
Add perfetto visualization
#1147 opened Feb 25, 2026 by copybara-service bot Loading…
@hgao327
Move ifrt based reshard out of experimental. Leaving intermediate resharding and sidechannel resharding in experimental.
#1146 opened Feb 24, 2026 by copybara-service bot Loading…
@hgao327
[Tunix] Update qwix dependency and TPU test image.
#1145 opened Feb 24, 2026 by copybara-service bot Loading…
@hgao327
speedup trainer for RL
#1141 opened Feb 24, 2026 by copybara-service bot Loading…
@hgao327
plug in simple print export
#1139 opened Feb 23, 2026 by copybara-service bot Loading…
@hgao327
Fix sharding rank mismatch for LoRA weights and optimizer states
#1138 opened Feb 23, 2026 by selamw1 Loading…
1
Sanbao/gpt
#1137 opened Feb 23, 2026 by susanbao Loading…
6 tasks
@hgao327
3
Add DeepSWE train script and add configurable prompt key.
#1134 opened Feb 23, 2026 by copybara-service bot Loading…
@hgao327
add metric agentic instrumentations
#1132 opened Feb 20, 2026 by copybara-service bot Loading…
remat MLP block
#1130 opened Feb 20, 2026 by copybara-service bot Loading…
[Tunix] Connects compute_logps_micro_batch_size to ref inference.
#1127 opened Feb 19, 2026 by copybara-service bot Loading…
remove unnecessary example filtering
#1125 opened Feb 19, 2026 by copybara-service bot Loading…
Demo: Scaling Agentic Workflows with GKE Agent Sandboxes
#1123 opened Feb 19, 2026 by igooch • Draft
6
[Tunix] Remove utils.time_measure from the training loop.
#1118 opened Feb 19, 2026 by copybara-service bot Loading…
[Tunix] Reset vLLM prefix cache after updating parameters.
#1117 opened Feb 19, 2026 by copybara-service bot Loading…
Add VLM SFT example
#1104 opened Feb 17, 2026 by copybara-service bot Loading…
Allow sampler to take in images
#1103 opened Feb 16, 2026 by copybara-service bot Loading…
Internal test
#1101 opened Feb 15, 2026 by copybara-service bot Loading…
expert parallelism config in base rollout
#1099 opened Feb 13, 2026 by khatwanimohit Loading…
6 tasks done
7
Add memory optimization options to state transfer.
#1098 opened Feb 13, 2026 by copybara-service bot Loading…
Adding dtype casting for direct state transfer.
#1097 opened Feb 13, 2026 by NicoGrande Loading…
6 tasks done
6
create clear boundaries for global step, mini batch and micro batch
#1096 opened Feb 13, 2026 by copybara-service bot Loading…
Expose enable_thinking in RLCluster.generate.
#1087 opened Feb 12, 2026 by copybara-service bot Loading…
ProTip! Add no:assignee to see everything that’s not assigned.