Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
227 Open 4,117 Closed
227 Open 4,117 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

Introduce an inflight computation throttler to rate-limit TPU computations in MaxText training engine.
#4629 opened Jul 27, 2026 by copybara-service Bot Loading…
@SurbhiJainUSC
1
Fix the error message in test_scan_layers_mismatch_tpu
#4628 opened Jul 27, 2026 by xibinliu Collaborator Loading…
4 tasks done
3
Add rope_pairwise config flag to optimize YaRN RoPE
#4627 opened Jul 27, 2026 by dandragona Collaborator Loading…
3 tasks done
2
Optimize TPU Tokamax ring backward
#4626 opened Jul 27, 2026 by huytransformer Collaborator • Draft
4 tasks
2
feat(input): Megatron MMap Data Formats for MaxText Grain
#4625 opened Jul 27, 2026 by Yitrus Loading…
4 tasks done
1
fix: MTP CP-aware roll, segment-aware roll, and synthetic packing support
#4623 opened Jul 27, 2026 by chiaotung97 Loading…
6 tasks done
2
Support packed TPU Tokamax ring attention
#4622 opened Jul 27, 2026 by huytransformer Collaborator • Draft
4 tasks
2
Restore Engram defaults in RLConfig
#4621 opened Jul 27, 2026 by csgoogle Loading…
3
fix(nnx): Fix Linen checkpoint restoration into NNX Gemma4 model
#4619 opened Jul 27, 2026 by hsuan-lun-chiang Collaborator • Draft
1
Support vllm_decode for Qwen3-VL
#4618 opened Jul 27, 2026 by hengtaoguo Collaborator • Draft
4 tasks
2
fix: fallback to dot_product for autoselected attention on CPU to prevent Tokamax NotImplementedError
#4617 opened Jul 27, 2026 by chiajunglien Contributor Loading…
4 tasks done
5
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
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
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
ProTip! Filter pull requests by the default branch with base:main.