Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
162 Open 3,742 Closed
162 Open 3,742 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

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
[DEBUG DO NOT REVIEW] reproduce CI issues with additional print statements
#4157 opened Jun 12, 2026 by igorts-git Collaborator • Draft
4 tasks
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
9
extract_answer: prefer boxed{N} extraction, fall back to legacy tags
#4150 opened Jun 11, 2026 by py4 Collaborator Loading…
4 tasks done
1
Add reward_functions_path + reward_functions CLI knobs for custom rewards
#4149 opened Jun 11, 2026 by py4 Collaborator Loading…
5 tasks done
1
Update multimodal docs with new models
#4148 opened Jun 11, 2026 by hengtaoguo Collaborator Loading…
4 tasks done
20
[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
Use regular checkout instead of github action checkout v5.
#4142 opened Jun 11, 2026 by wstcliyu Collaborator Loading…
4 tasks done
4
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
Fix step-1 NaN: run gemma attention + DeepSeek sparse-indexer logits in fp32
#4136 opened Jun 10, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
Make the distillation trainer enforce its own step count.
#4135 opened Jun 10, 2026 by copybara-service Bot Loading…
1
fix: update nnx decoders from main
#4132 opened Jun 10, 2026 by mesakhcienet Collaborator Loading…
4 tasks done
3
Add option to enable on-demand profiling server in MaxText ML Diagnostics
#4131 opened Jun 10, 2026 by copybara-service Bot Loading…
1
1
Make GCP region configurable for managed ML diagnostics runs in MaxText
#4130 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
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.