Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
192 Open 2,679 Closed
192 Open 2,679 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

[WIP] Integrate DS V32 with flash attention
#3087 opened Feb 4, 2026 by RissyRan Loading…
4 tasks
CI test: divide to 2 worker groups for more UT
#3086 opened Feb 4, 2026 by charlesli640 Loading…
4 tasks done
1
Add MROPE support. pull ready
#3085 opened Feb 4, 2026 by entrpn Loading…
4 tasks done
1
Fix b/481469708
#3084 opened Feb 4, 2026 by parambole • Draft
4 tasks
1
Fix checkpoint restore sharding error in tpu_integration_test pull ready
#3082 opened Feb 4, 2026 by hengtaoguo Loading…
4 tasks done
2
Support input buffer count for gmm
#3081 opened Feb 4, 2026 by BirdsOfAFthr Loading…
3 tasks done
2
Dump activation shardings draft Draft PR
#3080 opened Feb 4, 2026 by charlesli640 Loading…
4 tasks done
2
internal change
#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
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
11
Remove DPO (Direct Preference Optimization) feature
#3064 opened Feb 2, 2026 by ecnal-cienet • Draft
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
Move src/MaxText/layers to src/maxtext/layers
#3054 opened Jan 30, 2026 by dipannita08 Loading…
4 tasks done
2
Add wandb support
#3053 opened Jan 30, 2026 by Zephyr271828 Loading…
4 tasks done
@dipannita08
1
Refactor multimodal utils and migrate to src/maxtext gemini-review
#3052 opened Jan 30, 2026 by hengtaoguo Loading…
4 tasks done
8
move src/MaxText/configs to src/maxtext/configs
#3044 opened Jan 29, 2026 by khatwanimohit Loading…
4 tasks done
11
ProTip! What’s not been updated in a month: updated:<2026-01-04.