Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
164 Open 3,682 Closed
164 Open 3,682 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

Robust list config parsing and improved SFT docs
#4101 opened Jun 6, 2026 by shralex Collaborator Loading…
4 tasks done
1
Add links to Checkpoint Conversion Guide in getting started tutorials
#4100 opened Jun 6, 2026 by shralex Collaborator Loading…
4 tasks done
1
Clarify and improve scan_layers mismatch error handling
#4099 opened Jun 6, 2026 by shralex Collaborator Loading…
4 tasks done
1
Clarify GCS bucket permission requirements in SFT and RL tutorials
#4098 opened Jun 6, 2026 by shralex Collaborator Loading…
4 tasks done
Add validation to block unsupported parameters in post-training SFT, …
#4097 opened Jun 6, 2026 by shralex Collaborator Loading…
4 tasks done
1
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
Update geting started doc
#4094 opened Jun 6, 2026 by SurbhiJainUSC Collaborator Loading…
4 tasks done
[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
Fix elastic data loading process count divisibility error in MaxText.
#4088 opened Jun 5, 2026 by copybara-service Bot Loading…
@lukebaumann
1
Complete Deprecation of Legacy AQT & Transition to Qwix/FP8
#4087 opened Jun 5, 2026 by sarunsingla11722 Collaborator Loading…
6 tasks done
1
Skip ring of experts all gather when EP = 1
#4086 opened Jun 5, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks done
1
Added live slice count
#4085 opened Jun 5, 2026 by ycchenzheng Collaborator Loading…
4 tasks done
@ycchenzheng @DannyLiCom
1
Add colocated python Dockerfile and build upload script
#4084 opened Jun 5, 2026 by ycchenzheng Collaborator Loading…
4 tasks done
@ycchenzheng
2
Add mean_reward to evaluate() via reusing user-provided reward functions
#4083 opened Jun 5, 2026 by py4 Collaborator Loading…
4 tasks done
2
Remove masks and update gmm function in moe.py gemini-review
#4082 opened Jun 5, 2026 by NuojCheng Collaborator Loading…
4 tasks done
3
Refactor train_loop into training_loop_iteration with explicit state dicts
#4079 opened Jun 5, 2026 by lukebaumann Collaborator Loading…
1
Snehalv dsv4 conv utility
#4078 opened Jun 5, 2026 by snehalv2002 Collaborator • Draft
4 tasks
chore: add type annotations to grpo_utils based on docstrings
#4076 opened Jun 5, 2026 by harshaa765 Loading…
3 tasks
fix: remove accidental debug logging from decoder pipeline
#4075 opened Jun 5, 2026 by harshaa765 Loading…
2 tasks
1
Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer.
#4073 opened Jun 5, 2026 by copybara-service Bot Loading…
1
Fix MaxTextCheckpointManager pytype compilation error
#4072 opened Jun 4, 2026 by copybara-service Bot Loading…
@bvandermoon
1
[WIP-Experimental]Onboard qwen1.5-moe-a2.7b model in MaxText
#4070 opened Jun 4, 2026 by YixuanWang-99 Collaborator Loading…
4 tasks
ProTip! Updated in the last three days: updated:>2026-06-04.