Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
191 Open 2,598 Closed
191 Open 2,598 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

Rename sharding_names to out_sharding in NNX Variable metadata
#3001 opened Jan 22, 2026 by copybara-service bot Loading…
@cgarciae
1
PR 3000! draft Draft PR
#3000 opened Jan 22, 2026 by NuojCheng • Draft
4 tasks
1
Update tflops calc for qwen3 next & GatedDeltaNet
#2999 opened Jan 22, 2026 by Rohan-Bierneni Loading…
4 tasks done
@Rohan-Bierneni
1
Add jaxpr dump support
#2998 opened Jan 22, 2026 by NuojCheng Loading…
4 tasks done
1
Update the Ironwood offloading description pull ready
#2997 opened Jan 22, 2026 by RissyRan Loading…
4 tasks done
5
Add NNX model support with Linen/NNX checkpoint interoperability
#2996 opened Jan 22, 2026 by ecnal-cienet • Draft
4 tasks done
1
Add Deepseek and GPT OSS weight mapping for RL.
#2995 opened Jan 22, 2026 by abhinavclemson Loading…
4 tasks done
2
Fix: ValueError: filedescriptor out of range in select() for papermill notebook execution
#2994 opened Jan 22, 2026 by RexBearIU Loading…
4 tasks done
1
Comment out run_apidoc call in Sphinx conf.py
#2993 opened Jan 22, 2026 by copybara-service bot Loading…
@shralex
1
Test new deps
#2992 opened Jan 22, 2026 by kanglant • Draft
4 tasks done
1
test
#2991 opened Jan 22, 2026 by RissyRan Loading…
4 tasks
Move src/MaxText/sft to src/maxtext/trainers/post_train/sft gemini-review
#2988 opened Jan 21, 2026 by SurbhiJainUSC Loading…
4 tasks done
7
Add TE NVFP4 quantization config without RHT
#2987 opened Jan 21, 2026 by hx89 Loading…
4 tasks
Force layout on Q for MLA.
#2986 opened Jan 21, 2026 by copybara-service bot Loading…
1
Print logic axes
#2985 opened Jan 21, 2026 by Shuang-cnt • Draft
4 tasks done
11
[src/MaxText/inference/offline_engine.py] Make OfflineEngine a fluent interface
#2983 opened Jan 21, 2026 by SamuelMarks Loading…
4 tasks done
1
[src/MaxText/input_pipeline/_tfds_data_processing.py] Builder Pattern for TFDS Preprocessing Pipeline: preprocessing_pipeline
#2982 opened Jan 21, 2026 by SamuelMarks Loading…
4 tasks done
1
[src/MaxText/input_pipeline/_hf_data_processing.py] Refactor to fluent interface for constructing Grain pipelines (introduce GrainPipelineBuilder); increase type-safety
#2981 opened Jan 21, 2026 by SamuelMarks Loading…
4 tasks done
1
NNX migration: NNX utils and unit tests
#2980 opened Jan 21, 2026 by xibinliu • Draft
4 tasks done
1
Add flops calculation for DeepSeek v3.2
#2979 opened Jan 21, 2026 by RissyRan Loading…
4 tasks done
@shuningjin @gagika
@gobbleturk @NuojCheng
1
Adding chat template to vllm decode.
#2978 opened Jan 20, 2026 by NicoGrande Loading…
4 tasks done
4
Tyagiva diagon upgrade pull ready
#2976 opened Jan 20, 2026 by tyagivaji Loading…
4 tasks done
4
Record maxtext & post-training requirements commit ids to CI workflow gemini-review
#2975 opened Jan 20, 2026 by SurbhiJainUSC Loading…
4 tasks done
@A9isha
12
checkpoint utility: shard checkpoint, monitor peak
#2974 opened Jan 20, 2026 by shuningjin Loading…
4 tasks done
@RissyRan @hengtaoguo @NicoGrande
3
Add Qwen3-Next to checkpoint util
#2973 opened Jan 20, 2026 by Rohan-Bierneni Loading…
4 tasks done
@Rohan-Bierneni
1
ProTip! Filter pull requests by the default branch with base:main.