Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
158 Open 3,757 Closed
158 Open 3,757 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

Introduce SubBatchCheckpointManager interface.
#4171 opened Jun 15, 2026 by copybara-service Bot Loading…
1
Refactor moe.p: gmm and a2a unsort
#4170 opened Jun 15, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks done
@Shuwen-Fang
4
Add support for keep_every_nth_step in checkpointing options.
#4169 opened Jun 15, 2026 by copybara-service Bot Loading…
1
Add weight for ragged gather kernel and enable fan out in bwd ragged sort
#4166 opened Jun 15, 2026 by NuojCheng Collaborator Loading…
4 tasks done
1
Add ep as dp logical rule for scaling
#4163 opened Jun 15, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
Add on-the-fly dynamic SafeTensors loading support and remove redundant tensor handling logic
#4162 opened Jun 15, 2026 by copybara-service Bot Loading…
@abhinavclemson
1
Configure gemini-investigate on build failure for UploadDockerImages.yml
#4161 opened Jun 13, 2026 by shralex Collaborator Loading…
4 tasks done
1
Add low-memory streaming conversion for unscanned DeepSeek-family checkpoints
#4160 opened Jun 13, 2026 by discobot Loading…
3 of 4 tasks
1
3
Reverts changelist 926917348
#4159 opened Jun 12, 2026 by copybara-service Bot Loading…
1
Update Gemma3 multimodal SFT Jupyter notebook
#4154 opened Jun 11, 2026 by SurbhiJainUSC Collaborator Loading…
4 tasks done
2
[DeepSeek-V4] Implement model integration, decoders, and configuration stack gemini-review
#4153 opened Jun 11, 2026 by parambole Collaborator Loading…
4 tasks done
33
extract_answer: prefer boxed{N} extraction, fall back to legacy tags
#4150 opened Jun 11, 2026 by py4 Collaborator Loading…
4 tasks done
2
Add reward_functions_path + reward_functions CLI knobs for custom rewards
#4149 opened Jun 11, 2026 by py4 Collaborator Loading…
5 tasks done
1
[DO NOT MERGE] Feat/nnx set defaults true g3 test pull ready
#4146 opened Jun 11, 2026 by ecnal-cienet Collaborator Loading…
4 tasks done
1
Update google-cloud-mldiagnostics to >=1.0.3
#4144 opened Jun 11, 2026 by copybara-service Bot Loading…
1
Disable group offset for tokamax gmm
#4143 opened Jun 11, 2026 by NuojCheng Collaborator Loading…
4 tasks done
1
Bump the pip group across 3 directories with 5 updates dependencies Pull requests that update a dependency file python Pull requests that update python code
#4139 opened Jun 10, 2026 by dependabot Bot Loading…
1
# Description
#4138 opened Jun 10, 2026 by copybara-service Bot Loading…
4 tasks done
@lukebaumann
1
Make the distillation trainer enforce its own step count.
#4135 opened Jun 10, 2026 by copybara-service Bot Loading…
1
add ragged sort support to batch split
#4127 opened Jun 10, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
Blockwise parallel distillation in the scan mode
#4126 opened Jun 10, 2026 by vlad-karp Collaborator Loading…
4 tasks done
1
Add local_sa_* flags for local attention tuning
#4125 opened Jun 10, 2026 by aireenmei Collaborator • Draft
4 tasks
Fix misleading preemption logs in checkpointing
#4124 opened Jun 9, 2026 by lukebaumann Collaborator Loading…
1
Migrate Tunix's SFT Checkpoint Manager usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#4116 opened Jun 9, 2026 by copybara-service Bot Loading…
1
Add orbax-checkpoint to post-training GitHub dependencies.
#4115 opened Jun 9, 2026 by copybara-service Bot Loading…
1
ProTip! Add no:assignee to see everything that’s not assigned.