Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
195 Open 2,500 Closed
195 Open 2,500 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

Fix the LIBTPU_INIT_ARGS logging issue
#2905 opened Dec 30, 2025 by xibinliu Loading…
4 tasks done
1
Update docker image tags in GCR
#2904 opened Dec 30, 2025 by SurbhiJainUSC Loading…
4 tasks done
1
Add a script to verify safetensor checkpoint gemini-review
#2903 opened Dec 30, 2025 by RissyRan Loading…
4 tasks done
@hengtaoguo @shuningjin
10
test script for deepseek gemini-review
#2902 opened Dec 30, 2025 by shuningjin Loading…
4 tasks done
@parambole @RissyRan
19
Soft distillation training support
#2901 opened Dec 30, 2025 by gagika • Draft
4 tasks done
1
Fix the AddLabel workflow permission issue
#2899 opened Dec 29, 2025 by xibinliu Loading…
4 tasks done
2
Small updates on MoE configuration gemini-review
#2895 opened Dec 29, 2025 by RissyRan Loading…
4 tasks done
@richjames0
6
Fix Pydantic validation error for Vertex Tensorboard args
#2893 opened Dec 29, 2025 by hsuan-lun-chiang Loading…
4 tasks done
3
Add fractional batch size search for MaxText estimator draft Draft PR
#2891 opened Dec 27, 2025 by NuojCheng • Draft
4 tasks
1
Feat: Update RL on Multi-Host TPUs tutorial for clarity and structure
#2890 opened Dec 24, 2025 by RexBearIU Loading…
4 tasks done
15
Fix Codecov setup
#2889 opened Dec 24, 2025 by shralex Loading…
4 tasks done
1
Posttraining documentation refine
#2888 opened Dec 24, 2025 by hsuan-lun-chiang Loading…
4 tasks done
11
fix: add missing module init and update docs
#2887 opened Dec 24, 2025 by mesakhcienet • Draft
4 tasks done
Implement Best Fit Packing Algorithm in Grain Pipeline for Reduced Padding
#2886 opened Dec 24, 2025 by bzantium Loading…
4 tasks done
1
1
feat: migrate pipeline to nnx
#2885 opened Dec 24, 2025 by mesakhcienet • Draft
4 tasks
Implement Warmup-Stable-Decay (WSD) Learning Rate Schedule
#2883 opened Dec 24, 2025 by bzantium Loading…
4 tasks done
1
1
feat: support qwen3 next with muon
#2875 opened Dec 23, 2025 by Prayer3th Loading…
4 tasks done
@parambole @shuningjin @Rohan-Bierneni
6
feat: support KDA
#2874 opened Dec 23, 2025 by jimoosciuc Loading…
4 tasks done
@parambole @RissyRan @Rohan-Bierneni
[DECOUPLED-MODE] Adding Decoupling Logic
#2865 opened Dec 21, 2025 by gulsumgudukbay Loading…
4 tasks done
Add automatic API documentation generation
#2862 opened Dec 19, 2025 by melissawm Loading…
4 tasks done
Adding supported models to post_training_index.md pull ready
#2860 opened Dec 19, 2025 by jacoguzo Loading…
4 tasks done
1
Jacoguzo/tutorials editable code pull ready
#2856 opened Dec 18, 2025 by jacoguzo Loading…
4 tasks done
2
Fix two issues that blocks training loop with continuous checkpoint enabled.
#2854 opened Dec 17, 2025 by copybara-service bot Loading…
reduce tfds packing intermediate batch_size
#2847 opened Dec 16, 2025 by aireenmei • Draft
4 tasks
Docs: Enhance landing pages for Run, Guides, and Reference
#2839 opened Dec 16, 2025 by RobMulla Loading…
4 tasks done
1
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.