Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
172 Open 3,107 Closed
172 Open 3,107 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

Add open-r1/OpenR1-Math-220k dataset to RL
#3509 opened Mar 26, 2026 by SurbhiJainUSC • Draft
4 tasks done
1
As seen below:
#3508 opened Mar 26, 2026 by copybara-service bot Loading…
1
Add time stamp for GA test
#3507 opened Mar 26, 2026 by charlesli640 Loading…
4 tasks done
1
Bringup Qwen2.5-1.5B gemini-review
#3506 opened Mar 26, 2026 by ChingTsai Loading…
4 tasks done
5
Add link checker to docs build workflow
#3504 opened Mar 25, 2026 by SurbhiJainUSC • Draft
4 tasks done
2
Add time info through user warning for debugging purpose
#3502 opened Mar 25, 2026 by charlesli640 Loading…
4 tasks done
1
DO NOT MERGE Clean check vma pull ready
#3501 opened Mar 25, 2026 by Shuwen-Fang Loading…
4 tasks
1
NNX migration prep (3/N): TrainState, model creation, and end-to-end training loop
#3500 opened Mar 25, 2026 by ecnal-cienet • Draft
4 tasks done
1
Re-enable pre-commit insert-license hook (another implementation).
#3499 opened Mar 25, 2026 by igorts-git • Draft
4 tasks
1
Change quantization_local_shard_count to number of slices
#3496 opened Mar 25, 2026 by RUEI4341 Loading…
4 tasks done
Add checkpoint deletion options to configuration and checkpoint manager
#3494 opened Mar 24, 2026 by copybara-service bot Loading…
1
[DRAFT] Paged Stashing for ring-of-experts MoE Activation Memory
#3493 opened Mar 24, 2026 by abhinavgoel95 • Draft
4 tasks
10
Support option to skip the optimizer for training step gemini-review
#3490 opened Mar 24, 2026 by RissyRan Loading…
4 tasks done
@aireenmei @gagika @gobbleturk
9
[Test] Rope change
#3486 opened Mar 23, 2026 by RissyRan • Draft
4 tasks
1
Add option to replicate attn weights on FSDP or EP
#3480 opened Mar 21, 2026 by gobbleturk Loading…
1
Support modern uv project management (uv init + uv add)
#3474 opened Mar 20, 2026 by igorts-git • Draft
4 tasks done
1
1
Fix DiLoCo training compatibility issues
#3471 opened Mar 20, 2026 by khatwanimohit Loading…
4 tasks done
5
NNX migration prep (2/N): NNX utils and sharding utilities
#3470 opened Mar 20, 2026 by ecnal-cienet • Draft
4 tasks done
1
Remove legacy uppercase MaxText from src, to ensure maxtext works on MacOS which has a case-insensitive filesystem.
#3467 opened Mar 20, 2026 by syrkis Loading…
4 tasks done
1
Fix src/MaxText references in GPU/runner Dockerfiles pull ready
#3462 opened Mar 20, 2026 by bvandermoon Loading…
4 tasks done
1
Integrating b/v blocking for linear softmax cross entropy loss in maxtext.
#3461 opened Mar 19, 2026 by copybara-service bot Loading…
1
maxtext/tunix lora integration [WIP]
#3453 opened Mar 18, 2026 by andytwigg • Draft
4 tasks
1
Add simplified APIs for model obtaining maxtext models
#3450 opened Mar 18, 2026 by A9isha Loading…
4 tasks done
1
modify process_data to generate separate user/system parts in prompts
#3445 opened Mar 18, 2026 by andytwigg Loading…
4 tasks done
4
NNX train
#3442 opened Mar 18, 2026 by charlesli640 • Draft
4 tasks done
3
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.