Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
230 Open 4,121 Closed
230 Open 4,121 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

[MaxText][RL] Omit Redundant Weight Resharding Prior to Step 0 Evaluation & Add Engram Inheritance
#4636 opened Jul 27, 2026 by igorts-git Collaborator • Draft
4 tasks done
1
experiment
#4635 opened Jul 27, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks
1
Fix DeepSeek decode running the prefill attention path under NNX
#4634 opened Jul 27, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
2
Refactor layer-wise tests with torch
#4633 opened Jul 27, 2026 by hengtaoguo Collaborator • Draft
4 tasks done
2
Add ep to moe test for chunking
#4631 opened Jul 27, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks done
2
Add flag-controlled sliced MLA projections
#4630 opened Jul 27, 2026 by dandragona Collaborator Loading…
4 tasks done
2
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
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 Loading…
4 tasks done
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…
4 tasks done
@niting
1
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…
ProTip! no:milestone will show everything without a milestone.