Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
218 Open 4,299 Closed
218 Open 4,299 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

fix(ci): Align Airflow E2E DAG parameter names with ml-auto-solutions
#4806 opened Aug 10, 2026 by RexBearIU Collaborator Loading…
4 tasks done
1
Add GPT-OSS-20b E2E test scripts
#4805 opened Aug 10, 2026 by chiajunglien Collaborator Loading…
4 tasks done
11
Expose parallel_threads flag for HF conversion to prevent OOM
#4804 opened Aug 10, 2026 by chiajunglien Collaborator Loading…
4 tasks done
4
Fix read-only config mutation, mock target, and carry shape in TestApplyLayersSequentiallyMetadataAxisName
#4803 opened Aug 10, 2026 by copybara-service Bot Loading…
4 tasks done
@shralex
1
Checkpoint Validation Pipeline Operations: Airflow polling integrations, PyConfig overrides, and pipeline lifecycle tooling
#4800 opened Aug 10, 2026 by olufiyin19 Collaborator Loading…
4 tasks done
12
Fix multihost_utils.process_allgather tiled=True in hf_utils.py
#4799 opened Aug 9, 2026 by olufiyin19 Collaborator Loading…
7
Add gdn_matmul_precision opt-in knob for Qwen3-Next Gated DeltaNet
#4795 opened Aug 8, 2026 by Noman654 Loading…
1
Gemma-4 E2B: enable image (multimodal) E2E parity — vision clipped-linears + padded-patch masking + PLE substitution (E4B vision-tower validated; E4B E2E pending)
#4790 opened Aug 8, 2026 by lokic233 Loading…
10
Add JAX compilation cache clear fixture to prevent XLA memory accumulation gemini-review
#4784 opened Aug 7, 2026 by SurbhiJainUSC Collaborator Loading…
4 tasks done
6
qwen3 muon fork
#4779 opened Aug 7, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks
5
[maxtext] Add block-diffusion pre-training gemini-review
#4776 opened Aug 7, 2026 by ethannnnnn Collaborator Loading…
4 tasks done
4
Fix Qwen 3.5 35B RL gibberish output issue and show end-to-end RL convergence gemini-review
#4770 opened Aug 7, 2026 by AntonyMei Collaborator Loading…
16
Support Ring Attention with DeepSeek DSA Sparse Indexer gemini-review
#4767 opened Aug 6, 2026 by zcjhao Collaborator Loading…
4 tasks done
23
Support gemma4 decoder under explicit sharding mode
#4766 opened Aug 6, 2026 by csgoogle Contributor Loading…
1 of 2 tasks
4
Moe dynamic redundancy
#4760 opened Aug 6, 2026 by snehalv2002 Collaborator • Draft
4 tasks
7
[DiLoCo] Streaming diloco with vmap (SPMD Version) enhancement New feature or request gemini-review
#4759 opened Aug 6, 2026 by Dr-Left Collaborator Loading…
4 tasks done
@khatwanimohit
30
Add Grain support for multimodal SFT data processing pipeline
#4754 opened Aug 6, 2026 by snehalv2002 Collaborator Loading…
4 tasks done
9
Download tokenizers from GCS in unit tests instead of HF pull ready
#4752 opened Aug 5, 2026 by igorts-git Collaborator Loading…
4 tasks done
10
Dsv4 pm
#4751 opened Aug 5, 2026 by snehalv2002 Collaborator • Draft
4 tasks
5
fix(pylint): ignore datasets module to resolve E0611 no-name-in-module errors
#4750 opened Aug 5, 2026 by NuojCheng Collaborator Loading…
3
Add README to maxtext.trainers gemini-review
#4749 opened Aug 5, 2026 by melissawm Collaborator Loading…
4 tasks done
3
[No merge] Draft test pull ready
#4747 opened Aug 5, 2026 by NuojCheng Collaborator • Draft
4 tasks
Decouple vision & decoder block routing in multimodal processor pull ready
#4746 opened Aug 5, 2026 by subawocit Collaborator Loading…
4 tasks done
10
Honor partial_rotary_factor on the MRoPE path
#4742 opened Aug 5, 2026 by adityasingh2400 Loading…
1
1
perf(GMM-v2): fuse transpose_rhs into GMM-v2 kernel for DLHS path
#4741 opened Aug 5, 2026 by Insideyyy Collaborator Loading…
4 tasks done
@RissyRan @shuningjin
1
ProTip! Filter pull requests by the default branch with base:main.