Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
224 Open 4,108 Closed
224 Open 4,108 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

Intercept and promote single-device arrays before checkpointing in MaxText.
#4614 opened Jul 26, 2026 by copybara-service Bot Loading…
@lukebaumann
1
Fix eval and checkpointing under load-balanced CP
#4613 opened Jul 26, 2026 by huytransformer Collaborator • Draft
4 tasks
2
Support MLA TPU Tokamax ring attention
#4610 opened Jul 25, 2026 by huytransformer Collaborator • Draft
4 tasks
2
Fix roundtrip checkpoint conversion test
#4609 opened Jul 24, 2026 by hengtaoguo Collaborator Loading…
4 tasks done
3
End2End Video SFT with Varied Size/Duration
#4608 opened Jul 24, 2026 by hengtaoguo Collaborator • Draft
4 tasks
2
Fix logical to physical axis mapping for attention heads.
#4607 opened Jul 24, 2026 by copybara-service Bot Loading…
@niting
1
Clean up splash attention imports gemini-review pull ready
#4606 opened Jul 24, 2026 by huytransformer Collaborator Loading…
4 tasks done
5
Add axis inference logic and disable EP flags when EP is not active
#4604 opened Jul 24, 2026 by NuojCheng Collaborator • Draft
4 tasks
2
Bump the pip group across 2 directories with 4 updates dependencies Pull requests that update a dependency file python Pull requests that update python code
#4603 opened Jul 24, 2026 by dependabot Bot Loading…
1
Fix test_scan_layers_mismatch_tpu layer path assertion
#4602 opened Jul 24, 2026 by hsuan-lun-chiang Collaborator • Draft
4 tasks done
2
feat: MTP CP-aware roll, segment-aware roll, and synthetic packing support
#4600 opened Jul 24, 2026 by chiaotung97 Loading…
6 tasks done
2
Fix test_scan_layers_mismatch_tpu layer path assertion
#4599 opened Jul 24, 2026 by hsuan-lun-chiang Collaborator • Draft
4 tasks done
2
maxtext checkpoint implementation explained
#4595 opened Jul 24, 2026 by xibinliu Collaborator • Draft
4 tasks done
1
Enable experimental V quantization, quantization strategies.
#4594 opened Jul 23, 2026 by copybara-service Bot Loading…
Add custom MLP vision projector and tests
#4593 opened Jul 23, 2026 by subawocit Collaborator Loading…
4 tasks done
3
[maxtext] Wire opt-in block-diffusion GRPO
#4592 opened Jul 23, 2026 by ethannnnnn Loading…
2
[maxtext] Add exact block-diffusion policy replay
#4591 opened Jul 23, 2026 by ethannnnnn Loading…
2
[maxtext] Export exact stochastic denoising traces
#4590 opened Jul 23, 2026 by ethannnnnn Loading…
2
[maxtext] Add model-aware block-diffusion OPD
#4589 opened Jul 23, 2026 by ethannnnnn Loading…
2
[maxtext] Add reusable block-diffusion rollout transitions
#4588 opened Jul 23, 2026 by ethannnnnn Loading…
2
[maxtext] Add generalized block-diffusion CFT and SFT
#4587 opened Jul 23, 2026 by ethannnnnn Loading…
2
[maxtext] Add generalized block-causal attention
#4586 opened Jul 23, 2026 by ethannnnnn Loading…
2
Fix eval_start_step logic with eval_interval
#4585 opened Jul 23, 2026 by zxhe-sean Collaborator Loading…
4 tasks done
2
Eliminate redundant sorts in BWD pass by caching indexes between embedding chunks
#4583 opened Jul 23, 2026 by Shuwen-Fang Collaborator Loading…
3 tasks done
5
Add RL GRPO gemma4-26b tutorial and fix doc conversion links
#4580 opened Jul 23, 2026 by dasoto Collaborator Loading…
4 tasks done
3
ProTip! What’s not been updated in a month: updated:<2026-06-26.