Skip to content

Pull requests: AI-Hypercomputer/maxtext

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

Streaming diloco with vmap finished
#4287 opened Jun 27, 2026 by Dr-Left Collaborator • Draft
4 tasks done
1
Revised Add dynamic Hugging Face SafeTensors loading to MaxText
#4286 opened Jun 26, 2026 by copybara-service Bot Loading…
@abhinavclemson
1
[DO NOT REVIEW]Enable LoRA training in the NNX path of MaxText (pre-training and native SFT)
#4284 opened Jun 26, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks done
1
Internal Change
#4283 opened Jun 26, 2026 by copybara-service Bot Loading…
1
Upgrade pretraining to jax 0.10.2
#4282 opened Jun 26, 2026 by gobbleturk Collaborator Loading…
4 tasks
1
Add llama3.1-70b E2E test scripts for pre-training
#4279 opened Jun 26, 2026 by chiajunglien • Draft
4 tasks done
3
Add gemma4-26b E2E test scripts for pre-training
#4278 opened Jun 26, 2026 by chiajunglien • Draft
4 tasks done
2
docs: improve lora guide and demo notebook
#4277 opened Jun 26, 2026 by RexBearIU Collaborator Loading…
4 tasks done
[CI] Split tpu unit tests into 2 parallel worker groups
#4276 opened Jun 26, 2026 by xibinliu Collaborator Loading…
4 tasks done
7
zjiahao/optimize reorder sequence
#4275 opened Jun 25, 2026 by JHCuc3m Collaborator • Draft
3 of 4 tasks
1
Added missing activation checkpointing for DeepseekV4 model gemini-review
#4272 opened Jun 25, 2026 by dipakg-lang 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
#4271 opened Jun 25, 2026 by dependabot Bot Loading…
1
Resolve step timing misattribution and isolate checkpoint overhead in MaxText.
#4270 opened Jun 25, 2026 by copybara-service Bot Loading…
4 tasks done
1
feat(lora): save/restore LoRA config in checkpoint metadata
#4269 opened Jun 25, 2026 by RexBearIU Collaborator Loading…
4 tasks done
11
Fix/nnx linen parity gaps unit test
#4267 opened Jun 25, 2026 by hsuan-lun-chiang Collaborator • Draft
4 tasks
2
Add Tokamax ring attention for TPU
#4266 opened Jun 25, 2026 by huytransformer Collaborator Loading…
4 tasks done
1
Fix ragged gather reduce correctness pull ready
#4262 opened Jun 24, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
Optimize MLA generate_mask with scatter update
#4261 opened Jun 24, 2026 by JHCuc3m Collaborator • Draft
2 of 4 tasks
1
Ubench combined testing draft Draft PR
#4256 opened Jun 24, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
fix: add missing header all-gather
#4254 opened Jun 24, 2026 by mesakhcienet Collaborator • Draft
4 tasks
1
Mesa/nnx set defaults true
#4253 opened Jun 24, 2026 by mesakhcienet Collaborator • Draft
4 tasks
1
[RL] Fix shape mismatch on tail batch in GRPO training
#4252 opened Jun 24, 2026 by susanbao Collaborator Loading…
4 tasks done
2
Fix config validation test issues in maxtext.
#4248 opened Jun 24, 2026 by copybara-service Bot Loading…
@lukebaumann
1
Implement pathways-based in-process snapshot elasticity in MaxText.
#4247 opened Jun 24, 2026 by copybara-service Bot Loading…
@lukebaumann
1
Upgrade ragged_gather_reduce SparseCore kernel to match vLLM v2
#4246 opened Jun 24, 2026 by copybara-service Bot Loading…
@gobbleturk
1
ProTip! Follow long discussions with comments:>50.