Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
229 Open 2,399 Closed
229 Open 2,399 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

Update documentation links and remove old requirements.
#2828 opened Dec 14, 2025 by copybara-service bot Loading…
@shralex
Update configuration doc
#2827 opened Dec 13, 2025 by RissyRan Loading…
4 tasks done
Add citation entry for repository
#2825 opened Dec 13, 2025 by shuningjin Loading…
4 tasks done
6
RemoteIterator skip iter ckpt
#2824 opened Dec 12, 2025 by aireenmei • Draft
4 tasks done
Docs: Add Google Analytics ID to sphinx configuration
#2823 opened Dec 12, 2025 by RobMulla Loading…
2
Refactor dockerfiles and deprecate stable_stack, jax_ai_image, custom wheels mode
#2822 opened Dec 12, 2025 by SurbhiJainUSC • Draft
4 tasks done
Adding cards to the index pages of RtD
#2821 opened Dec 12, 2025 by jacoguzo Loading…
4 tasks done
Make the workflows more robust
#2820 opened Dec 11, 2025 by xibinliu Loading…
4 tasks done
[WIP] checkpoint util: gpt-oss, hf -> orbax
#2818 opened Dec 11, 2025 by shuningjin • Draft
4 tasks
DeepSeek V3.1 announcement & Update benchmark tooling script
#2817 opened Dec 11, 2025 by Rohan-Bierneni Loading…
4 tasks done
17
Fix typo in title for multi-host TPU tutorial
#2816 opened Dec 11, 2025 by melissawm Loading…
4 tasks done
[src/MaxText/{configs/{rl.yml,types.py},examples/rl_llama3_demo.ipynb,rl/train_rl.py}] Use pydantic natively in RL ; nest GRPO
#2815 opened Dec 11, 2025 by SamuelMarks Loading…
4 tasks done
Fix broken links in docs
#2814 opened Dec 11, 2025 by hengtaoguo Loading…
4 tasks done
[DO NOT MERGE] Skip tests for doc-only PRs
#2810 opened Dec 10, 2025 by hengtaoguo • Draft
4 tasks done
Switch from pytest-split to pytest-xdist for parallel test execution.
#2808 opened Dec 10, 2025 by copybara-service bot Loading…
@shralex
Fix the type of micro_batch_size pull ready
#2807 opened Dec 10, 2025 by xuefgu Loading…
4 tasks done
Update documentation links and remove old requirements.
#2805 opened Dec 10, 2025 by copybara-service bot Loading…
@shralex
update links urls pull ready
#2804 opened Dec 10, 2025 by dcharrezt Loading…
4 tasks done
1
Add vLLM Support for GPT OSS and its mapping generator for tunix.
#2803 opened Dec 10, 2025 by abhinavclemson Loading…
4 tasks done
5
Docs: add ml-diagnostic guide
#2800 opened Dec 9, 2025 by Shuang-cnt Loading…
4 tasks done
12
Enable continuous checkpoint by default in MaxText.
#2799 opened Dec 9, 2025 by copybara-service bot Loading…
Fix HLO Dump
#2795 opened Dec 8, 2025 by gobbleturk • Draft
4 tasks
This change introduces a pure JAX implementation of flash attention to Maxtext, designed as a drop-in replacement for the existing Pallas kernel. In this cl we set up the stage by integrating it with maxtext in fsdp mode. We have plans for further optimizations to close the gap with pallas using different techniques such as:
#2793 opened Dec 5, 2025 by copybara-service bot Loading…
Migrate multi_token_prediction to NNX
#2788 opened Dec 4, 2025 by hsuan-lun-chiang • Draft
4 tasks
Fixes Issue #2781 for gcloud command in getting number of workers
#2782 opened Dec 3, 2025 by MrGeislinger Loading…
4 tasks done
1
ProTip! no:milestone will show everything without a milestone.