Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
156 Open 3,539 Closed
156 Open 3,539 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

Qwen3-30b learn to init
#3942 opened May 19, 2026 by gagika Collaborator Loading…
4 tasks done
1
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
1
Add num_trays_per_compute_node.
#3937 opened May 18, 2026 by copybara-service Bot Loading…
1
Add fp8_{e4m3, e5m2}.
#3936 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: auto-adapt Linen checkpoints on load (params + full-state)
#3929 opened May 18, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
Fix GPT3 attention missing KV cache initialization and handling pull ready
#3927 opened May 17, 2026 by shralex Collaborator Loading…
4 tasks done
5
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…
5
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
[Qwen3.5] Fix caching bug in GDN layer for autoregressive mode
#3907 opened May 14, 2026 by Rohan-Bierneni Collaborator Loading…
4 tasks done
@parambole @Rohan-Bierneni
1
Add documentation for running MaxText via gcluster.
#3906 opened May 14, 2026 by copybara-service Bot Loading…
Add Gemma 4 E2B / E4B (text) support to MaxText gemini-review
#3904 opened May 14, 2026 by gagika Collaborator Loading…
4 tasks done
26
update post training deps.
#3898 opened May 13, 2026 by NicoGrande Collaborator • Draft
4 tasks
Add grpc authentication to maxengine_server
#3897 opened May 13, 2026 by bvandermoon Collaborator Loading…
4 tasks done
1
ProTip! Add no:assignee to see everything that’s not assigned.