Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
161 Open 3,485 Closed
161 Open 3,485 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 eval code owners
#3892 opened May 13, 2026 by dipannita08 Collaborator Loading…
4 tasks done
1
Experimental skills directory
#3891 opened May 13, 2026 by dipannita08 Collaborator Loading…
4 tasks done
4
mHC optimization for expansion rate 4 (flag controlled) gemini-review
#3890 opened May 12, 2026 by dandragona-dev Collaborator Loading…
4 tasks done
1
Add nnx weight init logic when vocab tiling is enabled
#3889 opened May 12, 2026 by NuojCheng Collaborator • Draft
4 tasks done
1
Support pytest markers regardless of decorator order
#3888 opened May 12, 2026 by igorts-git Collaborator • Draft
4 tasks done
1
[DO NOT MERGE] Add marker propagation verification test suite (expected to fail)
#3887 opened May 12, 2026 by igorts-git Collaborator • Draft
4 tasks
1
Add OLMo-3 7B stage-1 pretraining scripts gemini-review
#3886 opened May 12, 2026 by gagika Collaborator Loading…
4 tasks done
6
Do not block Add Label due to skipped checkRun pull ready
#3885 opened May 12, 2026 by charlesli640 Collaborator Loading…
4 tasks done
1
Plumbing and core MoE logic for router replay
#3881 opened May 12, 2026 by xuefgu Collaborator • Draft
4 tasks done
4
Improve error message when tokenize_data config doesn't match dataset gemini-review
#3879 opened May 12, 2026 by aireenmei Collaborator Loading…
4 tasks done
7
Publish TPU pre-training and post-training stable docker images to cloud-tpu-images
#3878 opened May 11, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks done
4
enable aot identification test
#3876 opened May 11, 2026 by NuojCheng Collaborator Loading…
4 tasks done
1
write dequantization scripts for DeepSeek V4 FP4/FP8 weights
#3873 opened May 11, 2026 by snehalv2002 Collaborator Loading…
4 tasks done
1
Add zero1 aot support in train compile
#3872 opened May 11, 2026 by NuojCheng Collaborator Loading…
4 tasks done
1
Implement custom MoE HashRouter, TopKRouter, and sqrtsoftplus
#3871 opened May 11, 2026 by parambole Collaborator • Draft
4 tasks
1
Conditionally branch tokamax.ragged_dot calls based on use_manual_quantization
#3869 opened May 11, 2026 by zxhe-sean Collaborator Loading…
4 tasks done
3
DeepSeek V4 Integration
#3867 opened May 11, 2026 by parambole Collaborator • Draft
4 tasks
1
Implement DeepSeek-V4 Compressed Attention Layers
#3866 opened May 11, 2026 by parambole Collaborator • Draft
4 tasks
1
DeepSeek-V4 Core Primitives
#3865 opened May 11, 2026 by parambole Collaborator • Draft
4 tasks
1
[DeepSeek v3] Add grad mask and update MLA init gemini-review
#3864 opened May 10, 2026 by gagika Collaborator Loading…
4 tasks done
9
Enable Qwen3-Omni SFT on ChartQA
#3863 opened May 10, 2026 by hengtaoguo Collaborator • Draft
4 tasks
1
Optimize MaxText unit and integration test suite runtime gemini-review pull ready
#3860 opened May 9, 2026 by shralex Collaborator Loading…
4 tasks done
14
Update optimization docs and add TPU v7x guide
#3857 opened May 8, 2026 by jacoguzo Collaborator Loading…
4 tasks done
11
Update docker image guide
#3855 opened May 8, 2026 by melissawm Collaborator Loading…
1 task done
Update JAX to 0.10.0 for pre-training
#3854 opened May 8, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks done
1
ProTip! Add no:assignee to see everything that’s not assigned.