Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
169 Open 3,718 Closed
169 Open 3,718 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 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 • Draft
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
1
Move torch installation and remove safetensors from checkpoint conversion guides
#4141 opened Jun 10, 2026 by copybara-service Bot Loading…
@SurbhiJainUSC
Fix KeyError('question') in evaluate-time reward path
#4140 opened Jun 10, 2026 by py4 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
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 scan mode
#4126 opened Jun 10, 2026 by vlad-karp Collaborator • Draft
4 tasks
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
Optimize distillation perf for qwen3-30b (20%→26% MFU) and gpt-oss-20b (17%→19%) gemini-review
#4118 opened Jun 9, 2026 by gagika 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
Update model readme's with model support claim
#4108 opened Jun 8, 2026 by Rohan-Bierneni Collaborator Loading…
4 tasks done
Deprecate aqt keep deps
#4107 opened Jun 8, 2026 by sarunsingla11722 Collaborator Loading…
4 tasks done
1
ProTip! no:milestone will show everything without a milestone.