Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
222 Open 4,080 Closed
222 Open 4,080 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 dependency version
#4582 opened Jul 23, 2026 by Shuang-cnt Collaborator • Draft
4 tasks done
2
Fix pyconfig strict validation for multimodal keys in text-only runs
#4581 opened Jul 23, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks done
2
Add RL GRPO gemma4-26b tutorial and fix doc conversion links
#4580 opened Jul 23, 2026 by dasoto Collaborator Loading…
4 tasks done
3
Add custom MLP vision projector and tests
#4579 opened Jul 23, 2026 by subawocit Collaborator • Draft
4 tasks done
1
Updates to JAX flash attention kernel.
#4578 opened Jul 23, 2026 by copybara-service Bot Loading…
Optimize chunked delta rule inversion.
#4577 opened Jul 23, 2026 by h1t35h Loading…
4 tasks done
3
feat: Support pre-training and post-training callbacks in promote_docker_image workflow
#4576 opened Jul 23, 2026 by RexBearIU Collaborator Loading…
4 tasks done
2
ci: add test duration tracking workflow
#4575 opened Jul 23, 2026 by hsuan-lun-chiang Collaborator Loading…
4 tasks done
1
[Do not submit] test 5 pull ready
#4573 opened Jul 22, 2026 by aireenmei Collaborator • Draft
4 tasks done
1
fix v7x failure on fractional eval batch test
#4572 opened Jul 22, 2026 by NuojCheng Collaborator Loading…
4 tasks done
5
Remove all-to-all operation from TSP/CP in Attention Segmentation Mask
#4571 opened Jul 22, 2026 by dinodeep Collaborator Loading…
4 tasks done
2
Add Gemma 4 tokenizer to assets.
#4569 opened Jul 22, 2026 by copybara-service Bot Loading…
1
Deprecate get_logical_axis_rule_from_config gemini-review pull ready
#4568 opened Jul 22, 2026 by NuojCheng Collaborator Loading…
4 tasks done
7
Adapt inspect_checkpoint.py for Automated Ckpt Validation Pipeline (Agent Integration)
#4567 opened Jul 22, 2026 by olufiyin19 Collaborator Loading…
4 tasks done
3
Unify the NNX and Linen restore paths in load_state_if_possible
#4566 opened Jul 22, 2026 by ecnal-cienet Collaborator Loading…
4 tasks done
7
Update pathwaysutils version
#4565 opened Jul 22, 2026 by DannyLiCom Collaborator Loading…
4 tasks done
2
Implement QK attention head chunking in MLA to reduce HBM footprint gemini-review
#4564 opened Jul 22, 2026 by zcjhao Collaborator Loading…
48
Batched Video Encoding and Token Mask Propagation for Qwen3* pull ready
#4561 opened Jul 21, 2026 by hengtaoguo Collaborator Loading…
4 tasks done
4
Updates to JAX flash attention kernel.
#4559 opened Jul 21, 2026 by copybara-service Bot Loading…
feat(test): add modular E2E TPU testing scripts for GPTOSS-20B
#4556 opened Jul 21, 2026 by RexBearIU Collaborator • Draft
4 tasks done
1
fix(nnx): fix gpt3 paxml checkpoint conversion state map and layout
#4555 opened Jul 21, 2026 by hsuan-lun-chiang Collaborator • Draft
4 tasks done
1
Adding improved profiles for qwen3 models
#4553 opened Jul 21, 2026 by h1t35h Loading…
4 tasks done
Fix hardcoded grain_worker_count in HF data processing pipelines to prevent OOM while running Multimodal SFT via pathway
#4552 opened Jul 21, 2026 by chiajunglien Contributor Loading…
4 tasks done
3
apple envy moe implementation
#4551 opened Jul 21, 2026 by entrpn Collaborator Loading…
4 tasks done
Migrate the test script to use to_maxtext.py for supported models
#4541 opened Jul 20, 2026 by AusarYao28 Collaborator Loading…
4 tasks done
11
ProTip! Filter pull requests by the default branch with base:main.