Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
161 Open 3,869 Closed
161 Open 3,869 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

Fix load-balanced TPU Splash masks
#4298 opened Jun 29, 2026 by huytransformer Collaborator • Draft
4 tasks
Add README to inference submodule
#4297 opened Jun 29, 2026 by melissawm Collaborator Loading…
1 task done
1
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
1
Add TPU7x specific CI test workflows
#4294 opened Jun 29, 2026 by darisoy 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: 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
Add support for On-The-Fly Dynamic SafeTensors loading.
#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
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
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
ProTip! Add no:assignee to see everything that’s not assigned.