Skip to content

Pull requests: AI-Hypercomputer/maxtext

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

Update release notes for 0.2.3
#4145 opened Jun 11, 2026 by SurbhiJainUSC Collaborator Loading…
4 tasks done
2
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
3
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
Fix ValueError in RemoteIterator.save_state on scaled-down topologies.
#4138 opened Jun 10, 2026 by copybara-service Bot Loading…
@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
Bumps the pinned google-tunix commit to
#4135 opened Jun 10, 2026 by copybara-service Bot Loading…
1
[NNX] Fix multi-token-prediction (MTP) eval under pure_nnx gemini-review
#4133 opened Jun 10, 2026 by ecnal-cienet Collaborator Loading…
4 tasks done
6
fix: update nnx decoders from main
#4132 opened Jun 10, 2026 by mesakhcienet Collaborator • Draft
4 tasks
1
Add option to enable on-demand profiling server in MaxText ML Diagnostics
#4131 opened Jun 10, 2026 by copybara-service Bot Loading…
1
1
Add managed_mldiagnostics_region to MaxText configuration
#4130 opened Jun 10, 2026 by copybara-service Bot Loading…
1
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
Fix rank mismatch in MaxText synthetic data sharding
#4122 opened Jun 9, 2026 by lukebaumann Collaborator Loading…
4 tasks done
4
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
[WIP] Support token dropping for ring of experts EP using ragged sort kernels
#4113 opened Jun 9, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
Add a new goodput elastic utils function record_slice_state()
#4111 opened Jun 8, 2026 by YixuanWang-99 Collaborator Loading…
4 tasks done
1
Deprecate aqt keep deps
#4107 opened Jun 8, 2026 by sarunsingla11722 Collaborator Loading…
4 tasks done
1
chore: remove aqtp dependency for CI debugging (trigger PR)
#4105 opened Jun 8, 2026 by sarunsingla11722 Collaborator Loading…
4 tasks done
Add ragged all-to-all pipeline correctness coverage
#4104 opened Jun 8, 2026 by huytransformer Collaborator • Draft
4 tasks done
1
Test replica resize without checkpoint
#4103 opened Jun 8, 2026 by DannyLiCom Collaborator • Draft
4 tasks
2
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.