Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
175 Open 2,547 Closed
175 Open 2,547 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

[DO NO MERGE] Draft for sparse
#2933 opened Jan 13, 2026 by RissyRan • Draft
4 tasks
12
[DRAFT] Insert gathers draft Draft PR
#2932 opened Jan 13, 2026 by NuojCheng • Draft
4 tasks
1
Disable a flaky test
#2930 opened Jan 12, 2026 by xuefgu Loading…
4 tasks done
7
Refactor model loading in distillation to remove Tunix adapter.
#2927 opened Jan 10, 2026 by gagika Loading…
4 tasks done
@richjames0 @sheng-li
1
Fix: Update Fine-tuning Tutorial and Structure
#2925 opened Jan 9, 2026 by Shuang-cnt Loading…
4 tasks done
19
Trigger unit tests for docker images upload
#2924 opened Jan 9, 2026 by xibinliu Loading…
4 tasks done
1
Add jupyter notebook tests to Github CI tests
#2921 opened Jan 8, 2026 by SurbhiJainUSC Loading…
4 tasks done
7
[Diloco] Diloco trainer
#2920 opened Jan 8, 2026 by khatwanimohit • Draft
4 tasks done
12
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
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: update knowledge distill docs and implementation
#2887 opened Dec 24, 2025 by mesakhcienet Loading…
4 tasks done
5
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
16
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
30
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
ProTip! Mix and match filters to narrow down what you’re looking for.