Skip to content

Pull requests: AI-Hypercomputer/maxtext

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

Refactor checkpointing to use Orbax v1.
#4314 opened Jul 1, 2026 by copybara-service Bot Loading…
1
Fix HF token validation to support Hugging Face CLI cache
#4313 opened Jul 1, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks done
1
Refactor DeepSeek MoE/Engram scanned decoder tests to use 3 layers
#4312 opened Jul 1, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks done
1
Validate checkpoint manager for elastic training.
#4311 opened Jul 1, 2026 by copybara-service Bot Loading…
@lukebaumann
1
Fix HF model dtype fallback in forward_pass_logit_checker.py
#4310 opened Jun 30, 2026 by subawocit Collaborator • Draft
4 tasks done
1
Fix IndivisibleError in RemoteIteratorWrapper by using global mesh size.
#4309 opened Jun 30, 2026 by copybara-service Bot Loading…
@lukebaumann
1
Add use_te_comm_gemm_overlap: integrates TransformerEngine collective GEMMs for tensor sequence parallelism
#4307 opened Jun 30, 2026 by dinodeep Collaborator Loading…
4 tasks done
6
DRAFT pull ready
#4306 opened Jun 30, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
feat(scan_layers): verify scan_layers compatibility from checkpoint metadata
#4304 opened Jun 30, 2026 by RexBearIU Collaborator • Draft
4 tasks done
1
Split heavy test jobs across more workers
#4303 opened Jun 30, 2026 by hsuan-lun-chiang Collaborator • Draft
4 tasks
1
Optimize SwiGLU activation sharding: add explicit sharding constraint…
#4301 opened Jun 30, 2026 by JHCuc3m Collaborator • Draft
3 of 4 tasks
1
Fix load-balanced TPU Splash masks
#4298 opened Jun 29, 2026 by huytransformer Collaborator Loading…
4 tasks done
1
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
2
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 Loading…
4 tasks done
@Dr-Left
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
#4279 opened Jun 26, 2026 by chiajunglien • Draft
4 tasks done
3
Add gemma4-26b E2E test scripts
#4278 opened Jun 26, 2026 by chiajunglien • Draft
4 tasks done
2
ProTip! What’s not been updated in a month: updated:<2026-06-01.