- Notifications
You must be signed in to change notification settings - Fork473
Pull requests: AI-Hypercomputer/maxtext
Author
Uh oh!
There was an error while loading.Please reload this page.
Label
Uh oh!
There was an error while loading.Please reload this page.
Projects
Uh oh!
There was an error while loading.Please reload this page.
Milestones
Uh oh!
There was an error while loading.Please reload this page.
Reviews
Assignee
Assigned to nobodyLoading
Uh oh!
There was an error while loading.Please reload this page.
Sort
Pull requests list
fix: remove unused import of maxtext_google from train.py
#3200 openedFeb 20, 2026 bykryvokhyzhaLoading…
4 tasks done
PR #3189: Migrate MaxText/train.py to maxtext/trainers/pre_train/train.py
#3198 openedFeb 19, 2026 bybvandermoonLoading…
4 tasks done
Make
to_maxtext.py capable of running as a library to fixdemo_decoding.ipynb gemini-review #3197 openedFeb 19, 2026 byigorts-git • Draft
4 tasks done
remove tf dependency from tokenizer.py gemini-review
#3196 openedFeb 19, 2026 byaireenmeiLoading…
4 tasks done
Workflow to save top-k teacher logits in GCS to use in distillaiton
#3193 openedFeb 19, 2026 byajkv-googleLoading…
4 tasks done
fix: Gemma3 scan_layers=True checkpoint conversion param mapping
#3191 openedFeb 19, 2026 bykryvokhyzhaLoading…
4 tasks done
Migrate MaxText/train.py to maxtext/trainers/pre_train/train.py pull ready
#3189 openedFeb 19, 2026 bybvandermoonLoading…
4 tasks done
Remove unnecessary sharding logic for configuring Orbax single replica restore and broadcast.
#3187 openedFeb 18, 2026 bycopybara-servicebotLoading…
Support checkpointing TE quantizations with new remat policies
#3185 openedFeb 18, 2026 byhx89Loading…
Checkpoint conversion tool: Optimize to_maxtext & Onboard deepseek2/3/3.2
#3184 openedFeb 18, 2026 byshuningjin • Draft
4 tasks
Integrate Engram into custom model gemini-review
#3183 openedFeb 18, 2026 byRissyRanLoading…
4 tasks done
Refactor execution of deepseek sparse layers in Decoder to use pure JAX.
#3182 openedFeb 18, 2026 bycopybara-servicebotLoading…
Add external model bringup guide gemini-review
#3151 openedFeb 17, 2026 byRissyRanLoading…
4 tasks done
Reformat multi_token_prediction with pyink
#3150 openedFeb 17, 2026 bybvandermoonLoading…
4 tasks done
ProTip! Exclude everything labeled
bug with-label:bug.