Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
182 Open 2,529 Closed
182 Open 2,529 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 doc-only change detection to skip tests
#2922 opened Jan 8, 2026 by hengtaoguo • Draft
4 tasks done
1
Add jupyter notebook tests to Github CI tests
#2921 opened Jan 8, 2026 by SurbhiJainUSC • Draft
4 tasks done
1
[Diloco] Diloco trainer
#2920 opened Jan 8, 2026 by khatwanimohit • Draft
4 tasks done
10
Insert all gather per repeats for pipeline parallelism draft Draft PR
#2919 opened Jan 7, 2026 by NuojCheng • Draft
4 tasks
1
Happy New Year | Copyright 2023–2026 Google LLC
#2916 opened Jan 7, 2026 by SamuelMarks Loading…
4 tasks done
1
[src/MaxText/pyconfig.py] New initialize_pydantic_with_maxtext_config function ; [src/MaxText/examples/rl_llama3_demo.ipynb] Use new function
#2915 opened Jan 7, 2026 by SamuelMarks Loading…
4 tasks done
1
[Diloco] add diloco related utils
#2913 opened Jan 6, 2026 by khatwanimohit Loading…
4 tasks done
1
Add Packing Support for Context Parallelism (Ring Attention)
#2906 opened Dec 31, 2025 by kocchop Loading…
4 tasks done
14
Soft distillation training support
#2901 opened Dec 30, 2025 by gagika Loading…
4 tasks done
@richjames0 @sheng-li
@RissyRan @SurbhiJainUSC
2
Fix the AddLabel workflow permission issue
#2899 opened Dec 29, 2025 by xibinliu Loading…
4 tasks done
2
Add fractional batch size search for MaxText estimator draft Draft PR gemini-review
#2891 opened Dec 27, 2025 by NuojCheng • Draft
4 tasks done
4
Fix Codecov setup
#2889 opened Dec 24, 2025 by shralex Loading…
4 tasks done
1
fix: update knowledge distill docs and implementation
#2887 opened Dec 24, 2025 by mesakhcienet Loading…
4 tasks done
1
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
13
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
24
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
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
Docs streamline fixes
#2838 opened Dec 16, 2025 by RobMulla Loading…
4 tasks done
1
ProTip! Filter pull requests by the default branch with base:main.