Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
140 Open 3,437 Closed
140 Open 3,437 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

Add documentation skill for manual review
#3822 opened May 5, 2026 by melissawm Collaborator Loading…
3 tasks done
2
[NNX] NNX migration prep (7/N): NNX-native MaxEngine inference
#3821 opened May 5, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
Update libtpu to 0.0.38 in tpu-requirements.txt
#3820 opened May 5, 2026 by copybara-service Bot Loading…
@SurbhiJainUSC
1
Integrate MTP with batch-split config
#3817 opened May 5, 2026 by copybara-service Bot Loading…
@jesselu-google
1
Update GRPO and RL diagrams pull ready
#3815 opened May 5, 2026 by melissawm Collaborator Loading…
1 task done
@jacoguzo
2
[Draft] draft Draft PR
#3814 opened May 5, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
Refactor elastic_utils.live_devices to take elastic_enabled flag.
#3812 opened May 5, 2026 by copybara-service Bot Loading…
@abhinavclemson
1
feat: Add DeepSeek-V4 Token Compressor
#3810 opened May 4, 2026 by parambole Collaborator • Draft
4 tasks done
Allow sharding data files by row when data file count is too small
#3807 opened May 4, 2026 by aireenmei Collaborator • Draft
4 tasks
Add checkpoint resharding script for faster loading gemini-review
#3801 opened May 3, 2026 by shuningjin Collaborator Loading…
4 tasks done
@RissyRan @hengtaoguo
10
Add weekly linkchecker and report failures on issue
#3792 opened May 1, 2026 by melissawm Collaborator Loading…
3 tasks done
1
[WIP] Multimodal quality benchmark
#3784 opened Apr 30, 2026 by hengtaoguo Collaborator • Draft
4 tasks
Exclude RNG state when updating vllm with checkpoint for pre-rl evaluation
#3782 opened Apr 30, 2026 by SurbhiJainUSC Collaborator Loading…
4 tasks done
Feat/Support-Qwix-Quantization-For-NNX-Models
#3781 opened Apr 30, 2026 by hsuan-lun-chiang Collaborator • Draft
4 tasks
Support specifying tokamax gmm tile sizes in MaxText
#3779 opened Apr 29, 2026 by darisoy Collaborator Loading…
4 tasks done
1
Reorganize pre-training doc
#3778 opened Apr 29, 2026 by melissawm Collaborator Loading…
2 tasks done
9
Add unit tests for evaluate_rl.py
#3776 opened Apr 29, 2026 by SurbhiJainUSC Collaborator Loading…
4 tasks done
Remove local sort after ragged all-to-all
#3774 opened Apr 29, 2026 by copybara-service Bot Loading…
[NNX] NNX migration prep (6/N): NNX-native DPO
#3773 opened Apr 29, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
[Distillation] Layer-wise LTI gemini-review
#3769 opened Apr 29, 2026 by vlad-karp Collaborator Loading…
4 tasks done
14
[NNX] NNX migration prep (5/N): correctness fixes and feature enablements
#3766 opened Apr 28, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
CI/CD UT optimization
#3753 opened Apr 27, 2026 by charlesli640 Collaborator Loading…
4 tasks done
1
[RFC][Docs] Design for JAX-MCTS (Tree of Thoughts) Inference Strategy
#3751 opened Apr 27, 2026 by yipkingster • Draft
3 of 4 tasks
Add dataset type olmo_grain for AI2 OLMo numpy pretrain mixes gemini-review
#3749 opened Apr 26, 2026 by gagika Collaborator Loading…
4 tasks done
25
Replace hardcoded bucket names with generic ones/env variables
#3738 opened Apr 24, 2026 by melissawm Collaborator Loading…
1 task done
4
ProTip! Type g i on any issue or pull request to go back to the issue listing page.