Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
144 Open 3,458 Closed
144 Open 3,458 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

Simplify maybe_initialize_jax_distributed_system()
#3847 opened May 7, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks done
1
[NNX] NNX migration prep (4.6/N): Linen<->NNX checkpoint comparator
#3846 opened May 7, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
Update MaxText version to v0.2.2 and add release notes.
#3845 opened May 7, 2026 by copybara-service Bot Loading…
@SurbhiJainUSC
1
[NNX] NNX migration prep (9.5/N): NNX + AQT in MaxEngine + serve-mode reload + gpt3 prefill fix
#3844 opened May 7, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
[NNX] NNX migration prep (4.5/N): Linen<->NNX checkpoint converter
#3843 opened May 7, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
Improve eval metrics logging gemini-review
#3840 opened May 7, 2026 by aireenmei Collaborator Loading…
4 tasks done
14
[Qwen3.5] Onboard model to Checkpointing Util & Verify Correctness
#3839 opened May 7, 2026 by Rohan-Bierneni Collaborator Loading…
4 tasks done
1
[Test only] Maxtext side changes for parallel replica
#3837 opened May 7, 2026 by lepan-google Loading…
[NNX] NNX migration prep (9/N): NNX-aware QK-Clip + checkpoint utilities
#3836 opened May 7, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
Fix decoupled tests, refactor ICI parallelism override removal gemini-review
#3835 opened May 7, 2026 by gulsumgudukbay Collaborator Loading…
4 tasks done
DRAFT draft Draft PR
#3834 opened May 7, 2026 by NuojCheng Collaborator • Draft
4 tasks
Add explicit service cleanup and timeout to fix flaky pathways tests
#3831 opened May 6, 2026 by xuefgu Collaborator • Draft
4 tasks done
1
[DRAFT] draft Draft PR
#3830 opened May 6, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
Extending vllm adapter to support mm
#3827 opened May 6, 2026 by NicoGrande Collaborator • Draft
4 tasks
1
Elasticity events goodput logs
#3826 opened May 6, 2026 by dipannita08 Collaborator Loading…
4 tasks
1
[NNX] NNX migration prep (8/N): NNX native lora grpo
#3824 opened May 6, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
Add documentation skill for manual review
#3822 opened May 5, 2026 by melissawm Collaborator Loading…
3 tasks done
2
[NNX] NNX migration prep (7/N): NNX-native MaxEngine inference
#3821 opened May 5, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
Integrate MTP with batch-split config
#3817 opened May 5, 2026 by copybara-service Bot Loading…
@jesselu-google
1
[Draft] draft Draft PR
#3814 opened May 5, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
Refactor elastic_utils.live_devices to take elastic_enabled flag.
#3812 opened May 5, 2026 by copybara-service Bot Loading…
@abhinavclemson
1
Allow sharding data files by row when data file count is too small
#3807 opened May 4, 2026 by aireenmei Collaborator • Draft
4 tasks
Add weekly linkchecker and report failures on issue
#3792 opened May 1, 2026 by melissawm Collaborator Loading…
3 tasks done
1
[WIP] Multimodal quality benchmark
#3784 opened Apr 30, 2026 by hengtaoguo Collaborator • Draft
4 tasks
Feat/Support-Qwix-Quantization-For-NNX-Models
#3781 opened Apr 30, 2026 by hsuan-lun-chiang Collaborator • Draft
4 tasks
ProTip! What’s not been updated in a month: updated:<2026-04-08.