Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
239 Open 4,426 Closed
239 Open 4,426 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 support for custom GCS path to log ML Diagnostic telemetry data.
#4960 opened Aug 21, 2026 by copybara-service Bot Loading…
1
Fix vocab tiling hidden-state cotangent scaling
#4959 opened Aug 21, 2026 by JKSPL Loading…
4 tasks done
3
Chris/non spmd dlco
#4958 opened Aug 21, 2026 by Dr-Left Collaborator • Draft
6
Fix fp8 MoE on the sparse_matmul path
#4957 opened Aug 21, 2026 by gulsumgudukbay Collaborator Loading…
Fix quantized MoE on the dense_matmul path
#4955 opened Aug 20, 2026 by gulsumgudukbay Collaborator Loading…
4
feat: Auto-detect and convert Tunix post-training checkpoints on load
#4951 opened Aug 20, 2026 by hsuan-lun-chiang Collaborator • Draft
3 tasks done
7
fix: compute local block padding for dynamic splash indexer mask in CP
#4950 opened Aug 20, 2026 by RexBearIU Collaborator Loading…
2 of 4 tasks
2
allow grain to read tfds config
#4949 opened Aug 20, 2026 by aireenmei Collaborator Loading…
4 tasks done
8
Fix indexer mask padding in Tokamax Flash Attention with Context Parallelism
#4947 opened Aug 19, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks done
1
Fix NaN loss by using l2Norm in GDN v3 kernel
#4946 opened Aug 19, 2026 by muskansh-google Contributor • Draft
4 tasks
7
Support loading Tunix post-train checkpoints in MaxText pre-training
#4945 opened Aug 19, 2026 by hsuan-lun-chiang Collaborator • Draft
6
[DiLoCo] Add Docs for Streaming DiLoCo
#4942 opened Aug 19, 2026 by Dr-Left Collaborator Loading…
4 tasks done
@Dr-Left
8
[Gemma4] Fix the bug of use_qk_norm accidently set to False in Gemma 4 configs
#4938 opened Aug 18, 2026 by Dr-Left Collaborator Loading…
4 tasks done
@Dr-Left
8
Add a sharding-aware version of Muon to maxtext.
#4936 opened Aug 18, 2026 by copybara-service Bot Loading…
1
Add support for Tokamax GMM v2 heuristic tiling function.
#4935 opened Aug 18, 2026 by copybara-service Bot Loading…
@CaptainO5
Fix duplicate XLA compilation of jit(train_step)
#4934 opened Aug 18, 2026 by olupton Contributor Loading…
4 tasks done
5
Build the Qwen3-Next shared expert only when shared_experts > 0
#4930 opened Aug 18, 2026 by WandLZhang Loading…
3
Let splash attention exceed 524,288 tokens when packing is off
#4929 opened Aug 18, 2026 by WandLZhang Loading…
3
fix(checkpointing): raise RuntimeError on fatal checkpointing errors pull ready
#4927 opened Aug 18, 2026 by RexBearIU Collaborator Loading…
4 tasks done
7
Step0 reforward test scripts
#4925 opened Aug 18, 2026 by entrpn Collaborator Loading…
4 tasks
8
Add HCA Static Compilation for Splash Attention [Deepseek v4]
#4924 opened Aug 17, 2026 by octatrifan Collaborator • Draft
4 tasks done
12
Prepare MaxText release 0.2.4
#4923 opened Aug 17, 2026 by copybara-service Bot Loading…
@SurbhiJainUSC
2
cosmos3_nano model bring up (patch PR#4475)
#4920 opened Aug 17, 2026 by lydhr Collaborator Loading…
4 tasks done
2
Archive older news updates to shorten main list
#4919 opened Aug 17, 2026 by bvandermoon Collaborator Loading…
4 tasks done
1
DSV4: Add indexer_save_selection and indexer_sharded_topk for the CSA indexer
#4915 opened Aug 17, 2026 by systalyze-ai Loading…
4 tasks done
5
ProTip! Follow long discussions with comments:>50.