Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
174 Open 3,894 Closed
174 Open 3,894 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

bug fixes for deepseek4
#4337 opened Jul 2, 2026 by snehalv2002 Collaborator • Draft
4 tasks
1
checkpoint conversion support for deepseek v4
#4336 opened Jul 2, 2026 by snehalv2002 Collaborator • Draft
4 tasks
1
fix grain_worker_count max value
#4335 opened Jul 2, 2026 by aireenmei Collaborator Loading…
4 tasks done
1
Remove stale Tunix degenerate group masking config
#4334 opened Jul 2, 2026 by AntonyMei Collaborator Loading…
4 tasks done
1
1
Move data_processing_test to cpu
#4333 opened Jul 2, 2026 by aireenmei Collaborator • Draft
4 tasks done
1
Automated Code Change
#4332 opened Jul 1, 2026 by copybara-service Bot Loading…
1
Automated Code Change
#4331 opened Jul 1, 2026 by copybara-service Bot Loading…
1
add Qwen3 VL tests, update logit checker, and golden logit generator
#4330 opened Jul 1, 2026 by subawocit Collaborator • Draft
4 tasks done
1
Use synthetic data for scripts under src/maxtext/configs/tpu
#4329 opened Jul 1, 2026 by aireenmei Collaborator • Draft
4 tasks done
1
Attempt to Update to jax 0.10.1
#4328 opened Jul 1, 2026 by gobbleturk Collaborator Loading…
4 tasks
1
Avoid checkpoint work on skipped steps
#4326 opened Jul 1, 2026 by SujeethJinesh Collaborator Loading…
4 tasks done
1
Fix HF token validation to support Hugging Face CLI cache
#4325 opened Jul 1, 2026 by copybara-service Bot Loading…
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
#4324 opened Jul 1, 2026 by dependabot Bot Loading…
1
Add experimental attention quantization flags.
#4321 opened Jul 1, 2026 by copybara-service Bot Loading…
1
[Do Not Merge] Fixes in forward pass logits for V4
#4319 opened Jul 1, 2026 by Rohan-Bierneni Collaborator • Draft
4 tasks
1
ci: add ci fail-fast implementation plan
#4315 opened Jul 1, 2026 by mesakhcienet Collaborator Loading…
4 tasks done
1
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 pull ready
#4313 opened Jul 1, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks done
@Shuwen-Fang
3
Validate checkpoint manager for elastic training.
#4311 opened Jul 1, 2026 by copybara-service Bot Loading…
@lukebaumann
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
ProTip! Type g i on any issue or pull request to go back to the issue listing page.