Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
159 Open 3,546 Closed
159 Open 3,546 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 Multi-tiered Checkpointing Support to Pathways
#3952 opened May 20, 2026 by SujeethJinesh Collaborator Loading…
4 tasks done
1
Remove tf from post-training
#3951 opened May 20, 2026 by aireenmei Collaborator Loading…
4 tasks done
1
Apply chat templating for lm_eval mmlu post training checkpoints
#3950 opened May 20, 2026 by dipannita08 Collaborator Loading…
4 tasks done
1
Add option to use Tokamax attention API in MaxText.
#3949 opened May 19, 2026 by copybara-service Bot Loading…
1
Fix MaxText Paged Attention OOM and AOT state freezing
#3948 opened May 19, 2026 by darisoy Collaborator • Draft
4 tasks done
1
Fix TPU kernel OOB memory access and correctness issues
#3946 opened May 19, 2026 by darisoy Collaborator Loading…
4 tasks done
2
NNX migration
#3943 opened May 19, 2026 by hsuan-lun-chiang Collaborator • Draft
4 tasks
1
Qwen3-30b learn to init
#3942 opened May 19, 2026 by gagika Collaborator Loading…
4 tasks done
4
Deprecate cloud_tpu_diagnostics support and update requirements lockfile
#3940 opened May 19, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks done
1
Add gemma3-4b E2E test scripts for pre-training and post-training
#3939 opened May 18, 2026 by SurbhiJainUSC Collaborator Loading…
4 tasks done
4
Add num_trays_per_compute_node.
#3937 opened May 18, 2026 by copybara-service Bot Loading…
1
Rename save/load_pytree to save/load. Eliminate most user-facing "pytree" terminology in favor of "state" as a more specific term. Add deprecations.py for handling deprecated public functions.
#3934 opened May 18, 2026 by copybara-service Bot Loading…
1
Supports batching in vLLM offline inference script pull ready
#3932 opened May 18, 2026 by learning-to-play Collaborator Loading…
4 tasks done
@jrplatin
1
Refactor tensor handling and add dynamic SafeTensors loading.
#3930 opened May 18, 2026 by copybara-service Bot Loading…
@abhinavclemson
1
NNX: load_parameters_path adapter for Linen<->NNX checkpoints gemini-review
#3929 opened May 18, 2026 by ecnal-cienet Collaborator Loading…
4 tasks done
6
Gemma 2 Local/Global KV Cache Split
#3926 opened May 17, 2026 by learning-to-play Collaborator Loading…
1
Gemma and Gemma 2 Native Serving Registration
#3925 opened May 17, 2026 by learning-to-play Collaborator Loading…
1
Updates 'full' remat policy to use nothing_saveable
#3924 opened May 17, 2026 by learning-to-play Collaborator Loading…
1
Clean up YAML files: lowercase booleans, trailing whitespace, and yamllint integration gemini-review
#3923 opened May 15, 2026 by igorts-git Collaborator Loading…
4 tasks done
5
[ROCm] optimizations to reduce temp size prior to training starts
#3922 opened May 15, 2026 by cj401-amd Loading…
6
Nicogrande/add gdn support
#3917 opened May 14, 2026 by NicoGrande Collaborator • Draft
4 tasks
1
Add debug logging for remote iterator
#3916 opened May 14, 2026 by aireenmei Collaborator Loading…
4 tasks done
2
[PROTOTYPE] E2E airflow tests workflow
#3915 opened May 14, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks
4
Deprecate AQT quantization in MaxText
#3912 opened May 14, 2026 by sarunsingla11722 Collaborator Loading…
4 tasks done
2
Add MaxText automated release preparation skill
#3910 opened May 14, 2026 by igorts-git Collaborator • Draft
4 tasks done
ProTip! Adding no:label will show everything without a label.