Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
186 Open 2,641 Closed
186 Open 2,641 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

Change mosaic_group_id in case of GPUs/CPUs
#3041 opened Jan 29, 2026 by Steboss Loading…
3 of 4 tasks
1
Fix gradient accumulation in post training sft
#3040 opened Jan 29, 2026 by ChingTsai Loading…
4 tasks done
@ChingTsai
1
[WIP] mm_utils refactor
#3039 opened Jan 29, 2026 by hengtaoguo • Draft
4 tasks done
1
Qwen mrope cont
#3038 opened Jan 29, 2026 by entrpn Loading…
4 tasks done
1
Add support for more datasets for rl
#3037 opened Jan 28, 2026 by A9isha Loading…
4 tasks done
16
Update import path for inference_microbenchmark.
#3036 opened Jan 28, 2026 by copybara-service bot Loading…
@SurbhiJainUSC
1
use prod diagon instances gemini-review
#3035 opened Jan 28, 2026 by tyagivaji Loading…
3 tasks done
2
Expand sharding dump
#3034 opened Jan 28, 2026 by Shuang-cnt • Draft
3 tasks done
1
Add Gemini CLI assistant gemini-review pull ready
#3033 opened Jan 28, 2026 by RissyRan Loading…
4 tasks done
68
Move src/MaxText/kernels to src/maxtext/kernels
#3032 opened Jan 28, 2026 by dipannita08 Loading…
4 tasks done
6
add google cloud mldiagnostics 0.5.10
#3031 opened Jan 28, 2026 by NuojCheng • Draft
4 tasks
1
Update run_maxtext_via_xpk.md to fix path to docker image build scripts
#3030 opened Jan 28, 2026 by dasoto Loading…
4 tasks done
Fix API documentation generation
#3029 opened Jan 28, 2026 by melissawm Loading…
4 tasks done
2
Exclude the yanked package in the requirement file and fix the curren…
#3027 opened Jan 28, 2026 by kanglant Loading…
4 tasks done
1
Move utilities to src/maxtext/utils
#3026 opened Jan 28, 2026 by SurbhiJainUSC • Draft
4 tasks done
1
NNX TrainState and unit tests
#3024 opened Jan 28, 2026 by xibinliu Loading…
4 tasks done
1
NNX migration: NNX utils
#3023 opened Jan 28, 2026 by xibinliu Loading…
4 tasks done
1
Adding new sharding and performance args for vLLM
#3020 opened Jan 27, 2026 by NicoGrande • Draft
4 tasks
1
Make pure-JAX batchsplit version of DeepSeekMoELayer.
#3018 opened Jan 27, 2026 by copybara-service bot Loading…
1
Update rl dependency
#3017 opened Jan 27, 2026 by NuojCheng • Draft
4 tasks
1
Move src/MaxText/assets to src/maxtext/assets/tokenizers pull ready
#3015 opened Jan 27, 2026 by SurbhiJainUSC Loading…
4 tasks done
2
Add Unit Test for NNX Decoder
#3014 opened Jan 27, 2026 by ecnal-cienet • Draft
4 tasks done
1
[DRAFT] do no merge
#3010 opened Jan 26, 2026 by shuningjin • Draft
4 tasks
1
UT: add UT for RL and Update Docs for RL single host
#3007 opened Jan 24, 2026 by charlesli640 Loading…
4 tasks done
1
Rename sharding_names to out_sharding in NNX Variable metadata
#3001 opened Jan 22, 2026 by copybara-service bot Loading…
@cgarciae
1
ProTip! no:milestone will show everything without a milestone.