Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
229 Open 4,481 Closed
229 Open 4,481 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

ci: Add tiered timeout-minutes across core PR CI workflows
#5006 opened Aug 26, 2026 by bvandermoon Collaborator Loading…
4 tasks done
2
Omni sft training gemini-review
#5005 opened Aug 25, 2026 by subawocit Collaborator • Draft
4 tasks done
14
chore: Upgrade MaxText post-training pinned dependencies to JAX 0.11.1
#5004 opened Aug 25, 2026 by darisoy Collaborator Loading…
7 tasks done
4
Integrate elastic goodput and slice efficiency into snapshotting
#5003 opened Aug 25, 2026 by dipannita08 Collaborator Loading…
4 tasks done
3
Dsv4 mtp support
#5002 opened Aug 25, 2026 by snehalv2002 Collaborator • Draft
4 tasks
7
Fix DeepSeek V4 sliding window attention mask for packed sequences
#5001 opened Aug 25, 2026 by snehalv2002 Collaborator • Draft
4 tasks
3
Dsv4 param mapping
#5000 opened Aug 25, 2026 by snehalv2002 Collaborator • Draft
4 tasks
8
Onboard explicit sharding to the Gemma models
#4996 opened Aug 25, 2026 by NuojCheng Collaborator Loading…
6 tasks done
10
Test: XLA:TPU host offload fails inside nested lax.scan
#4995 opened Aug 25, 2026 by NuojCheng Collaborator • Draft
4 tasks
3
[Draft] Bharatgen 221
#4994 opened Aug 25, 2026 by muskansh-google Contributor • Draft
4 tasks
5
Mask tiled_lhs on partial-K tile boundaries in _matmul (gmm_v2) so uninitialized/garbage padding rows in LHS don't propagate NaN*0 into the matmul accumulator.
#4993 opened Aug 25, 2026 by copybara-service Bot Loading…
Add explicit sharding support for the Qwen3 model family
#4992 opened Aug 25, 2026 by NuojCheng Collaborator Loading…
4 tasks done
4
fix(qwen3-next): make activation offload work under the nested block scan
#4990 opened Aug 25, 2026 by NuojCheng Collaborator Loading…
3
Add default rule constants to types.py and add base.yml parity test
#4985 opened Aug 24, 2026 by bvandermoon Collaborator Loading…
4 tasks done
7
Fix multiprocess dataloader CPU iteration leak via weakref sweep
#4983 opened Aug 24, 2026 by snehalv2002 Collaborator • Draft
4 tasks
3
Update load balance loss, bias term updates, and mlperf input pipeline
#4981 opened Aug 24, 2026 by copybara-service Bot Loading…
@RissyRan
1
Wire sharding-aware Muon optimizer and NamedSharding into training.
#4980 opened Aug 24, 2026 by copybara-service Bot Loading…
1
Add GPT-OSS-20b support at GPU.
#4978 opened Aug 24, 2026 by shurale-nkn Loading…
4 tasks done
Write post-training checkpoints in MaxText's on-disk layout
#4975 opened Aug 23, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
5
Register llama3-8b for HuggingFace conversion
#4974 opened Aug 23, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
4
Stop post-training tripping on config a model never set
#4973 opened Aug 23, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
4
Onboard GLM-5.2
#4972 opened Aug 23, 2026 by notabee Collaborator Loading…
4 tasks done
9
MoE fp8 backward: in-kernel drhs operand quantization + configurable gradient dtype
#4971 opened Aug 22, 2026 by ultrons Collaborator Loading…
7
MoE ring-of-experts: TC Pallas ring all-gather for the combine backward + x_sorted remat save
#4969 opened Aug 22, 2026 by ultrons Collaborator Loading…
5
Shard the GatedDeltaNet sequence under ici_context_parallelism
#4968 opened Aug 22, 2026 by WandLZhang Loading…
4 tasks done
1
2
ProTip! Mix and match filters to narrow down what you’re looking for.