Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
172 Open 3,798 Closed
172 Open 3,798 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

fix: add missing all-gather for nnx vocabulary tiling
#4236 opened Jun 23, 2026 by mesakhcienet Collaborator • Draft
4 tasks done
1
Qwen3 Coder 480B inference sharding changes for MaxText.
#4235 opened Jun 23, 2026 by copybara-service Bot Loading…
@learning-to-play
1
Enabled auxilillary loss free load balancing and sequence wise load b… gemini-review
#4233 opened Jun 22, 2026 by dipakg-lang Collaborator Loading…
1
Internal change
#4231 opened Jun 22, 2026 by copybara-service Bot Loading…
1
Enable packing for all-gather CP
#4230 opened Jun 22, 2026 by huytransformer Collaborator • Draft
4 tasks
1
Add cost estimates for ragged sort kernels pull ready
#4228 opened Jun 22, 2026 by NuojCheng Collaborator Loading…
4 tasks done
12
Fuse partial sum accumulation into TGMM kernel
#4227 opened Jun 22, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks done
1
Fix RL LR-schedule regression: default schedule length to max_train_steps
#4225 opened Jun 22, 2026 by py4 Collaborator Loading…
5 tasks done
2
[DRAFT] Workaround TPU SparseCore compilation failure in ring_ragged_unsort b…
#4223 opened Jun 22, 2026 by igorts-git Collaborator • Draft
4 tasks
1
Fix compile_cache_test by filtering cache files for jit_train_step
#4222 opened Jun 22, 2026 by igorts-git Collaborator Loading…
4 tasks done
1
Add dynamic loss balancing for distillation feature and logit losses
#4220 opened Jun 22, 2026 by ajkv-google Collaborator Loading…
4 tasks done
1
[Diloco] add DCN bandwidth throttling options
#4219 opened Jun 22, 2026 by khatwanimohit Collaborator Loading…
4 tasks done
10
Add support for On-The-Fly Dynamic SafeTensors loading.
#4218 opened Jun 22, 2026 by copybara-service Bot Loading…
@abhinavclemson
1
Fix/nnx sharding fix
#4216 opened Jun 22, 2026 by hsuan-lun-chiang Collaborator • Draft
4 tasks
1
Remove obsolete Tokamax gmm pipeline FSDP workaround
#4214 opened Jun 22, 2026 by huytransformer Collaborator Loading…
4 tasks done
1
Add ragged all-to-all pipeline correctness coverage
#4213 opened Jun 22, 2026 by huytransformer Collaborator Loading…
4 tasks done
1
Clarify RL checkpoint environment variables
#4212 opened Jun 21, 2026 by huytransformer Collaborator Loading…
4 tasks done
Use resolved mesh size for context parallel sharding
#4211 opened Jun 21, 2026 by huytransformer Collaborator Loading…
4 tasks done
4
Fix compile_cache_test to assert single jit_train_step cache file
#4210 opened Jun 20, 2026 by Liauuu Loading…
Fix logical sharding resolution in NNX
#4205 opened Jun 19, 2026 by xibinliu Collaborator Loading…
4 tasks done
13
Adding myself to codeowners to unblock my team on PRs.
#4201 opened Jun 18, 2026 by entrpn Collaborator Loading…
4 tasks done
1
feat(layers): implement custom shape-aligned attention and MoE primit…
#4200 opened Jun 18, 2026 by katyaoussar Loading…
Fix/nnx sharding
#4199 opened Jun 18, 2026 by hsuan-lun-chiang Collaborator • Draft
4 tasks
2
[ROCm]: test: update HLO references after tmem optimizations (PR4)
#4194 opened Jun 17, 2026 by cj401-amd Collaborator Loading…
1 task
1
[ROCm]: fix: reduce MoE temp memory — embedding cap, weight sum default, skip trivial specs (PR3)
#4193 opened Jun 17, 2026 by cj401-amd Collaborator Loading…
2 tasks
1
ProTip! Updated in the last three days: updated:>2026-06-20.