Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
156 Open 3,470 Closed
156 Open 3,470 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 zero1 aot support in train compile
#3872 opened May 11, 2026 by NuojCheng Collaborator • Draft
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
1
Fix ckpt conversion for qwen3-moe models (transformers==5.8.0)
#3868 opened May 11, 2026 by hengtaoguo 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
Refactor: extract shared post-training hooks and update SFT implementation
#3862 opened May 10, 2026 by igorts-git Collaborator Loading…
4 tasks done
3
Implement automated CI failure investigator gemini-review
#3861 opened May 9, 2026 by shralex Collaborator Loading…
4 tasks done
9
Optimize MaxText unit and integration test suite runtime
#3860 opened May 9, 2026 by shralex Collaborator Loading…
4 tasks done
1
Update optimization docs and add TPU v7x guide
#3857 opened May 8, 2026 by jacoguzo Collaborator Loading…
4 tasks done
8
Improve error message when tokenize_*_data config doesn't match dataset gemini-review
#3856 opened May 8, 2026 by aireenmei Collaborator Loading…
4 tasks done
7
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
Update First Run tutorial
#3853 opened May 8, 2026 by melissawm Collaborator Loading…
1 task done
Trigger tests using PR comments
#3850 opened May 8, 2026 by shralex Collaborator Loading…
4 tasks done
2
[NNX] NNX migration prep (10/N): vocab tiling custom_vjp with output-head carve-out
#3849 opened May 8, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
Simplify maybe_initialize_jax_distributed_system()
#3847 opened May 7, 2026 by SurbhiJainUSC Collaborator Loading…
4 tasks done
2
[NNX] NNX migration prep (4.6/N): Linen<->NNX checkpoint comparator
#3846 opened May 7, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
[NNX] NNX migration prep (9.5/N): NNX + AQT in MaxEngine + serve-mode reload + gpt3 prefill fix
#3844 opened May 7, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
[NNX] NNX migration prep (4.5/N): Linen<->NNX checkpoint converter
#3843 opened May 7, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
Improve eval metrics logging gemini-review
#3840 opened May 7, 2026 by aireenmei Collaborator Loading…
4 tasks done
15
[Qwen3.5] Onboard model to Checkpointing Util & Verify Correctness
#3839 opened May 7, 2026 by Rohan-Bierneni Collaborator Loading…
4 tasks done
1
ProTip! Exclude everything labeled bug with -label:bug.