Skip to content

Pull requests: AI-Hypercomputer/maxtext

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

Use tree reduction in dense gather reduce
#4296 opened Jun 29, 2026 by BirdsOfAFthr Collaborator Loading…
3 tasks done
1
DO NOT MERGE Chunk moe pull ready
#4295 opened Jun 29, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks
@Shuwen-Fang
Add TPU7x specific CI test workflows
#4294 opened Jun 29, 2026 by darisoy Collaborator Loading…
4 tasks done
1
Add Qwen3-VL-2B config
#4293 opened Jun 29, 2026 by subawocit Collaborator Loading…
4 tasks done
1
Internal Change
#4292 opened Jun 29, 2026 by copybara-service Bot Loading…
1
ci: add test duration tracking workflow
#4290 opened Jun 29, 2026 by hsuan-lun-chiang Collaborator • Draft
4 tasks
1
Fix TPU Splash local sliding window size
#4289 opened Jun 29, 2026 by huytransformer Collaborator Loading…
4 tasks done
4
fix: decoders and pipeline parity gaps of linen to nnx migrations
#4288 opened Jun 29, 2026 by mesakhcienet Collaborator Loading…
4 tasks done
2
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
[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
16
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
#4262 opened Jun 24, 2026 by NuojCheng Collaborator Loading…
4 tasks done
1
Optimize MLA generate_mask with scatter update
#4261 opened Jun 24, 2026 by JHCuc3m Collaborator • Draft
2 of 4 tasks
1
ProTip! Adding no:label will show everything without a label.