Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
106 Open 2,223 Closed
106 Open 2,223 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

Remove legacy MaxText references and clean up paths
#3757 opened Apr 27, 2026 by bvandermoon Collaborator Loading…
4 tasks done
3
Make node-limited routing the default in the DeepSeek3 batch-split.
#3756 opened Apr 27, 2026 by copybara-service Bot Loading…
@jesselu-google
1
CI/CD UT optimization
#3753 opened Apr 27, 2026 by charlesli640 Collaborator Loading…
4 tasks done
1
[Maxtext] Add a Pallas gather reduce kernel
#3752 opened Apr 27, 2026 by copybara-service Bot Loading…
1
[RFC][Docs] Design for JAX-MCTS (Tree of Thoughts) Inference Strategy
#3751 opened Apr 27, 2026 by yipkingster • Draft
3 of 4 tasks
Add dataset type olmo_grain for AI2 OLMo numpy pretrain mixes gemini-review
#3749 opened Apr 26, 2026 by gagika Collaborator Loading…
4 tasks done
3
Add distillation README and update xpk launcher gemini-review pull ready
#3748 opened Apr 26, 2026 by gagika Collaborator Loading…
4 tasks done
9
Add tokenizer support for Qwen3-30B-A3B-Base. pull ready
#3744 opened Apr 24, 2026 by niting Collaborator Loading…
4 tasks done
1
Internal change
#3743 opened Apr 24, 2026 by copybara-service Bot Loading…
@SurbhiJainUSC
1
Replace hardcoded bucket names with generic ones/env variables
#3738 opened Apr 24, 2026 by melissawm Collaborator Loading…
1 task done
Add fused_moe_mlp: fuse wi_0 and wi_1 into one grouped GEMM for MoE FFN1
#3736 opened Apr 23, 2026 by abhinavgoel95 Contributor Loading…
4 tasks done
5
Add support for flexible MTP layer architecture in MaxText.
#3734 opened Apr 23, 2026 by parambole Collaborator • Draft
4 tasks
1
make AddLabel workflow work as expected
#3733 opened Apr 23, 2026 by charlesli640 Collaborator Loading…
4 tasks done
12
Introduce cp-as-ep rule for long context training or strong scaling draft Draft PR
#3726 opened Apr 22, 2026 by NuojCheng Collaborator • Draft
4 tasks done
1
Add some log to debug AddLabel workflow stale Automatically applied to stale PRs.
#3723 opened Apr 22, 2026 by charlesli640 Collaborator • Draft
4 tasks done
1
Replace references to model_creation_utils.create_nnx_model.
#3720 opened Apr 22, 2026 by copybara-service Bot Loading…
4 tasks done
2
Move tokamax_group_sizes calculation inside tokamax branch.
#3715 opened Apr 22, 2026 by copybara-service Bot Loading…
1
HLO Deviation Unit Tests
#3713 opened Apr 21, 2026 by darisoy Collaborator Loading…
4 tasks done
27
[New Model Bringup] Initial Commit to enable Text-only architecture for Qwen3.5
#3712 opened Apr 21, 2026 by Rohan-Bierneni Collaborator Loading…
4 tasks done
4
Remove deprecated cross-program prefetch flags.
#3710 opened Apr 21, 2026 by copybara-service Bot Loading…
1
[Draft] Onboard ragged gather to moe draft Draft PR
#3707 opened Apr 21, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
draft pull ready
#3703 opened Apr 20, 2026 by shuningjin Collaborator • Draft
4 tasks done
1
Jackyf/qlora
#3702 opened Apr 20, 2026 by RexBearIU Collaborator • Draft
4 tasks
Speed up cpu-unit CI gemini-review
#3700 opened Apr 19, 2026 by gagika Collaborator • Draft
4 tasks
7
fix masking error when using mlp_bias=True causing NaN during gradien…
#3699 opened Apr 19, 2026 by snehalv2002 Collaborator Loading…
4 tasks done
4
ProTip! Exclude everything labeled bug with -label:bug.