Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
178 Open 3,821 Closed
178 Open 3,821 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

Add Tokamax ring attention for TPU
#4266 opened Jun 25, 2026 by huytransformer Collaborator • Draft
4 tasks
1
Add unit tests for Gemma 4 LoRA targeting and NNX caching
#4265 opened Jun 25, 2026 by RexBearIU Collaborator Loading…
4 tasks done
1
Expose sa_fuse_reciprocal and sa_use_base2_exp flags in MaxText
#4264 opened Jun 25, 2026 by dandragona Collaborator Loading…
4 tasks done
1
Add Qwen3-VL-4B support
#4263 opened Jun 24, 2026 by hengtaoguo Collaborator • Draft
4 tasks done
1
Fix ragged gather reduce correctness
#4262 opened Jun 24, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
Optimize MLA generate_mask with scatter update
#4261 opened Jun 24, 2026 by JHCuc3m Collaborator • Draft
2 of 4 tasks
1
Enable TC tiling on SparseCore and use num_lanes for row_subchunk_size. pull ready
#4257 opened Jun 24, 2026 by BirdsOfAFthr Collaborator Loading…
4 tasks done
1
Ubench combined testing draft Draft PR
#4256 opened Jun 24, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
NNX: fix Linen-parity gaps on the default path + unit tests
#4255 opened Jun 24, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
fix: add missing header all-gather
#4254 opened Jun 24, 2026 by mesakhcienet Collaborator • Draft
4 tasks
1
Mesa/nnx set defaults true
#4253 opened Jun 24, 2026 by mesakhcienet Collaborator • Draft
4 tasks
1
[RL] Fix shape mismatch on tail batch in GRPO training
#4252 opened Jun 24, 2026 by susanbao Collaborator Loading…
1
Fix config validation test issues in maxtext.
#4248 opened Jun 24, 2026 by copybara-service Bot Loading…
@lukebaumann
1
Implement pathways-based in-process snapshot elasticity in MaxText.
#4247 opened Jun 24, 2026 by copybara-service Bot Loading…
@lukebaumann
1
Upgrade ragged_gather_reduce SparseCore kernel to match vLLM v2
#4246 opened Jun 24, 2026 by copybara-service Bot Loading…
@gobbleturk
1
Implement checkpoint-based elasticity using set-based slice tracking in MaxText.
#4245 opened Jun 24, 2026 by copybara-service Bot Loading…
@lukebaumann
1
Add optional Approximate Top-K configuration for MLA Indexer
#4243 opened Jun 23, 2026 by JHCuc3m Collaborator Loading…
3 tasks done
5
Ragged gather reduce output in bf16 (previously f32)
#4240 opened Jun 23, 2026 by gobbleturk Collaborator Loading…
4 tasks
feat(agent): add checkpoint validation agent framework for qwen3-4b-u…
#4239 opened Jun 23, 2026 by olufiyin19 Collaborator Loading…
4 tasks
1
Add YaRN RoPE configuration support and enhance evaluation parameters pull ready
#4238 opened Jun 23, 2026 by JamesDeng42 Collaborator Loading…
4 tasks done
3
Nnx test 11 and vocab fix
#4237 opened Jun 23, 2026 by ecnal-cienet 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…
4 tasks done
1
Internal change
#4231 opened Jun 22, 2026 by copybara-service Bot Loading…
1
Fuse partial sum accumulation into TGMM kernel
#4227 opened Jun 22, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks done
@Shuwen-Fang
4
ProTip! Updated in the last three days: updated:>2026-06-21.