Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
166 Open 3,705 Closed
166 Open 3,705 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

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
3
docs: fix broken links and cross-references
#4121 opened Jun 9, 2026 by SurbhiJainUSC Collaborator Loading…
4 tasks done
1
Fix elastic manager initialization bug when elastic is disabled pull ready
#4119 opened Jun 9, 2026 by lukebaumann Collaborator Loading…
4 tasks done
1
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 usage of Orbax V0 Checkpoint Manager to V1 Checkpointer and. migrate distill_utils to v1 orbax semantics.
#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
Add ragged buffer to ring of experts
#4114 opened Jun 9, 2026 by RissyRan Collaborator • Draft
4 tasks
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
Revert "Fix the param names for Gemma3 model"
#4109 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
[Qwen3.5] Update model Readme and integrate XLML test scripts
#4106 opened Jun 8, 2026 by Rohan-Bierneni 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
Clarify and improve scan_layers mismatch error handling
#4099 opened Jun 6, 2026 by shralex Collaborator Loading…
4 tasks done
3
Add local_sa_block flags
#4096 opened Jun 6, 2026 by aireenmei Collaborator • Draft
4 tasks
1
<misc>: add scripts to run vanilla diloco on v5p cluster
#4095 opened Jun 6, 2026 by Dr-Left Collaborator • Draft
4 tasks
1
[WIP] Add checkpoint conversion agent utils
#4093 opened Jun 5, 2026 by YixuanWang-99 Collaborator Loading…
4 tasks
1
Fix: Eagerly initialize ML Diagnostics to prevent Protobuf descriptor collision crash
#4092 opened Jun 5, 2026 by bvandermoon Collaborator Loading…
4 tasks done
1
Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#4089 opened Jun 5, 2026 by copybara-service Bot Loading…
1
ProTip! Follow long discussions with comments:>50.