Skip to content

Pull requests: google/tunix

New pull request New
137 Open 964 Closed
137 Open 964 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

make perf engine (v1 or v2) and export function selectable.
#1191 opened Mar 3, 2026 by copybara-service bot Loading…
@sizhit2
copybara update for sampler
#1190 opened Mar 3, 2026 by copybara-service bot Loading…
@sizhit2
Extending scanned to unscanned support for GPT-OSS like models.
#1189 opened Mar 3, 2026 by NicoGrande • Draft
6 tasks done
@sizhit2
3
Enable pr from user's fork to auto assign issues correctly.
#1188 opened Mar 3, 2026 by copybara-service bot Loading…
[Tunix] Reduce log spam from type mismatch warnings.
#1187 opened Mar 3, 2026 by copybara-service bot Loading…
@sizhit2
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
1
split metric prefix
#1185 opened Mar 3, 2026 by copybara-service bot Loading…
@sizhit2
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
1
fix(rl): flush final partial micro-batch to avoid sample drop
#1179 opened Mar 1, 2026 by pjo256 Loading…
6 tasks done
4
Fix typo in documentation.
#1178 opened Mar 1, 2026 by copybara-service bot Loading…
@sizhit2
1
Implementation of Qwen3-VL
#1177 opened Feb 28, 2026 by ridcl Loading…
6 tasks done
1
8
my public commit msg
#1176 opened Feb 28, 2026 by copybara-service bot Loading…
@sizhit2
1
[Tunix] Add support for aligning 1D KV biases in sglang_jax.
#1175 opened Feb 28, 2026 by copybara-service bot Loading…
@sizhit2
[Tunix Perf] New perf tracer
#1172 opened Feb 27, 2026 by copybara-service bot Loading…
@sizhit2
[Tunix] update trajectory logging.
#1170 opened Feb 27, 2026 by copybara-service bot Loading…
@sizhit2
Add support for vllm sampler kwargs.
#1169 opened Feb 26, 2026 by NicoGrande Loading…
6 tasks done
30
[Tunix] Clear and reinitialize vLLM KV cache when loading weights.
#1167 opened Feb 26, 2026 by copybara-service bot Loading…
@sizhit2
fix: add thread safety locks to perf tracing module
#1166 opened Feb 26, 2026 by kbhujbal Loading…
6 tasks
2
test: add unit tests for utils/math_utils.py
#1165 opened Feb 26, 2026 by kbhujbal Loading…
6 tasks
4
Instrument agentic loop with perf v2
#1159 opened Feb 26, 2026 by copybara-service bot Loading…
@sizhit2
Instrument Tunix main with perf v2
#1158 opened Feb 26, 2026 by copybara-service bot Loading…
@sizhit2
[Tunix] Update qwix and vllm container
#1156 opened Feb 26, 2026 by copybara-service bot Loading…
@sizhit2
Add perfetto and logging export
#1155 opened Feb 25, 2026 by copybara-service bot Loading…
@hgao327
feat: add DFT loss and refactor SFT losses into submodule
#1153 opened Feb 25, 2026 by kryvokhyzha Loading…
5 of 6 tasks
5
Auto assign only one reviewer to the pull requests
#1150 opened Feb 25, 2026 by rajasekharporeddy Loading…
4 tasks done
1
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.