Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
177 Open 2,738 Closed
177 Open 2,738 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

Allow restore+broadcast logic to not require a single-replica sharding parameter, which is always constructed as a sharding over replica-local devices anyway.
#3132 opened Feb 13, 2026 by copybara-service bot Loading…
1
Set TF_CPP_MIN_LOG_LEVEL=0 in __init__.py
#3131 opened Feb 13, 2026 by igorts-git • Draft
4 tasks done
1
Do not store mtp_losses/mtp_acceptance in state params
#3130 opened Feb 12, 2026 by bvandermoon Loading…
4 tasks done
5
Updating config paths for MaxText on vLLM pull ready
#3129 opened Feb 12, 2026 by NicoGrande Loading…
4 tasks done
3
Disable/set tpu_only for tests required tpu backend
#3128 opened Feb 12, 2026 by NuojCheng • Draft
4 tasks
1
add cosine loss to train_distill
#3127 opened Feb 12, 2026 by entrpn Loading…
4 tasks done
[Draft] Engram integration
#3125 opened Feb 12, 2026 by RissyRan • Draft
1
restructure input_pipeline gemini-review pull ready
#3124 opened Feb 12, 2026 by aireenmei Loading…
4 tasks done
10
RL: Add chips_per_vm hyperparameter and fix an issue of missing config
#3123 opened Feb 11, 2026 by charlesli640 Loading…
4 tasks done
1
DOC: add a pre-commit hook to spell-check markdown files in docs/
#3122 opened Feb 11, 2026 by ev-br Loading…
4 tasks done
Integrate mHC with DeepSeek custom model gemini-review
#3115 opened Feb 9, 2026 by RissyRan Loading…
4 tasks done
@shuningjin @gagika
20
Feat/migrate decoder and tests to nnx
#3114 opened Feb 9, 2026 by hsuan-lun-chiang • Draft
4 tasks done
1
Add qwen2 implementation gemini-review
#3113 opened Feb 9, 2026 by ChingTsai Loading…
4 tasks done
10
Pathways replica resize
#3110 opened Feb 7, 2026 by lukebaumann • Draft
4 tasks
1
support openmath split and math_verify
#3106 opened Feb 6, 2026 by hengtaoguo • Draft
4 tasks
1
Trigger unit/integration tests for image build workflow
#3104 opened Feb 6, 2026 by xibinliu • Draft
4 tasks done
1
Add Qwen3 1.7B configs
#3103 opened Feb 6, 2026 by rxu183 Loading…
4 tasks done
4
Update deps pinned to jax version 0.8.2
#3098 opened Feb 5, 2026 by dipannita08 • Draft
4 tasks done
6
Fix duplicate peak learning rate in warmup schedule
#3095 opened Feb 5, 2026 by ChingTsai Loading…
4 tasks done
@ChingTsai
2
Test for generate_param_only_checkpoint_test
#3090 opened Feb 5, 2026 by hengtaoguo • Draft
4 tasks done
1
CI test: divide to 2 worker groups for more UT
#3086 opened Feb 4, 2026 by charlesli640 • Draft
4 tasks done
1
Dump activation shardings
#3080 opened Feb 4, 2026 by charlesli640 • Draft
4 tasks done
3
Roll forward after fix: https://github.com/AI-Hypercomputer/maxtext/pull/3050
#3079 opened Feb 4, 2026 by copybara-service bot Loading…
@aireenmei
1
Add Colocated Python Checkpointing
#3078 opened Feb 4, 2026 by SujeethJinesh • Draft
4 tasks
1
[Do Not Merge] Optimizations on Qwen3-Next GatedDeltaNet w/ Kernel & XProf Agent
#3077 opened Feb 4, 2026 by Rohan-Bierneni Loading…
4 tasks done
1
ProTip! Add no:assignee to see everything that’s not assigned.