Skip to content

Pull requests: AI-Hypercomputer/maxtext

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

Document enable_autocheckpoint and its distinction from emergency checkpointing.
#3537 opened Mar 31, 2026 by copybara-service bot Loading…
@abhinavclemson
Do not review, draft
#3534 opened Mar 31, 2026 by Shuwen-Fang Loading…
1
Use hf auth login instead of huggingface-cli login
#3533 opened Mar 31, 2026 by bvandermoon Loading…
4 tasks done
PR #3501: DO NOT MERGE Clean check vma
#3532 opened Mar 31, 2026 by Shuwen-Fang Loading…
4 tasks
1
vLLM based Eval framework gemini-review
#3531 opened Mar 31, 2026 by dipannita08 Loading…
4 tasks done
14
Add DeepSeek-V3 support for MaxText to Hugging Face conversion
#3530 opened Mar 31, 2026 by JamesDeng42 Loading…
4 tasks done
9
Enable Indexer cache for DS v3.2 decoding gemini-review
#3529 opened Mar 31, 2026 by RissyRan Loading…
4 tasks done
@shuningjin @Rohan-Bierneni
3
Fix Kimi-k2 checkpoint conversion
#3528 opened Mar 31, 2026 by gagika • Draft
4 tasks
1
Feat/nnx set defaults true
#3526 opened Mar 31, 2026 by ecnal-cienet • Draft
4 tasks done
1
NNX migration prep (4/N): sharding tools and Linen<->NNX checkpoint utilities
#3525 opened Mar 31, 2026 by ecnal-cienet • Draft
4 tasks done
1
[WIP, DO NOT REVIEW] Delete legacy DPO implementation
#3523 opened Mar 30, 2026 by igorts-git • Draft
4 tasks
1
Support Grain with Tfrecord in train_tokenizer
#3522 opened Mar 30, 2026 by aireenmei Loading…
4 tasks done
1
Shorten gradient accumulation test
#3521 opened Mar 30, 2026 by gobbleturk Loading…
4 tasks done
1
Add elastic pause/resume functionality to MaxText.
#3518 opened Mar 30, 2026 by copybara-service bot Loading…
@abhinavclemson
1
Adding gather reduce sparse core kernel.
#3517 opened Mar 28, 2026 by copybara-service bot Loading…
1
[WIP] MaxText RL with native vLLM
#3516 opened Mar 28, 2026 by hengtaoguo • Draft
4 tasks
1
[JAX] Replace jnp.clip(..., a_min=..., a_max=...) with jnp.clip(..., min=..., max=...).
#3512 opened Mar 27, 2026 by copybara-service bot Loading…
1
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
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
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.