Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
191 Open 2,691 Closed
191 Open 2,691 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 two issues that blocks training loop with continuous checkpoint enabled.
#3099 opened Feb 5, 2026 by copybara-service bot Loading…
1
Update deps pinned to jax version 0.8.2
#3098 opened Feb 5, 2026 by dipannita08 • Draft
4 tasks done
Qwen3-Omni Checkpoint Mapping
#3096 opened Feb 5, 2026 by entrpn Loading…
4 tasks done
1
Fix duplicate peak learning rate in warmup schedule
#3095 opened Feb 5, 2026 by ChingTsai Loading…
4 tasks done
2
Add support for overriding model architecture in Hugging Face conversion
#3094 opened Feb 5, 2026 by gagika Loading…
@entrpn @hengtaoguo
1
Test for generate_param_only_checkpoint_test
#3090 opened Feb 5, 2026 by hengtaoguo • Draft
4 tasks done
1
support epoch in hf pipeline pull ready
#3088 opened Feb 4, 2026 by aireenmei Loading…
4 tasks done
4
Integrate DeepSeek Sparse Attention with Tokamax Flash Attention gemini-review
#3087 opened Feb 4, 2026 by RissyRan Loading…
4 tasks done
@shuningjin @gagika
6
CI test: divide to 2 worker groups for more UT
#3086 opened Feb 4, 2026 by charlesli640 • Draft
4 tasks done
1
Fix b/481469708
#3084 opened Feb 4, 2026 by parambole • Draft
4 tasks
1
Dump activation shardings draft Draft PR
#3080 opened Feb 4, 2026 by charlesli640 • Draft
4 tasks done
2
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
[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
NNX Migration: all commits
#3075 opened Feb 3, 2026 by xibinliu • Draft
4 tasks done
1
Deepseek sharding for vLLM and MLA kernel plumbing
#3072 opened Feb 3, 2026 by khatwanimohit • Draft
4 tasks done
1
[DRAFT] SPMD PP gather weights and write custom vjp
#3071 opened Feb 3, 2026 by NuojCheng • Draft
4 tasks
1
Update TPU & GPU deps
#3070 opened Feb 3, 2026 by dipannita08 Loading…
4 tasks done
1
Make batch split factor configurable.
#3068 opened Feb 2, 2026 by copybara-service bot Loading…
1
DeepSeek MHC feature
#3065 opened Feb 2, 2026 by RissyRan Loading…
4 tasks done
39
Remove DPO (Direct Preference Optimization) feature
#3064 opened Feb 2, 2026 by ecnal-cienet Loading…
4 tasks done
3
[MaxEngine] Fix TypeError in prefill() during batched inference
#3063 opened Feb 2, 2026 by jaisong123 Loading…
1
Add Preprocessing and token placeholders
#3062 opened Feb 2, 2026 by hengtaoguo • Draft
4 tasks
support openmath split
#3059 opened Jan 31, 2026 by A9isha • Draft
4 tasks
1
Move maxtext integration
#3055 opened Jan 30, 2026 by dipannita08 Loading…
4 tasks done
1
ProTip! What’s not been updated in a month: updated:<2026-01-05.