Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
123 Open 995 Closed
123 Open 995 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add a basic prefill test in maxengine
#1227 opened Feb 1, 2025 by liurupeng Loading…
4 tasks done
1
Make float32_qk_product and float32_logits apply during inference
#1225 opened Feb 1, 2025 by philip-essential Loading…
4 tasks done
7
MaxText Trillium Benchmarking
#1224 opened Jan 31, 2025 by anfals • Draft
4 tasks
Maintain fp32 for optimizer state when offloading is enabled
#1223 opened Jan 31, 2025 by zhenying-liu Loading…
1
[NVIDIA] Support FP8 quantization for MOE layers
#1221 opened Jan 31, 2025 by kaixih Loading…
2
Add Pathways Benchmarking Recipes for Scale Testing (and fix bugs)
#1220 opened Jan 31, 2025 by SujeethJinesh • Draft
4 tasks
[DRAFT] Initial commit Maxtext unit tests with Pathways backend.
#1211 opened Jan 28, 2025 by RoshaniN Loading…
4 tasks
Mattdavidow ago rebased [DO NOT SUBMIT]
#1208 opened Jan 28, 2025 by gobbleturk • Draft
4 tasks
Mattdavidow ago refactor
#1207 opened Jan 28, 2025 by gobbleturk • Draft
4 tasks
Paged Attention
#1206 opened Jan 28, 2025 by patemotter Loading…
4 tasks done
Run Maxtext unit test on candidate JStS Image
#1204 opened Jan 27, 2025 by parambole Loading…
4 tasks done
add GCP platform check for workload observability
#1203 opened Jan 27, 2025 by jcyang43 • Draft
4 tasks done
[RFC]Prototype the MPMD GPipe-circular with single-host H100=8.
#1202 opened Jan 27, 2025 by lc5211 Loading…
1 task
2
Add 'model_name' label to metrics pull ready
#1195 opened Jan 23, 2025 by Bslabe123 Loading…
4 tasks done
WIP: Adding TransformerEngine support for Communication GeMM overlap for Tensor Parallelism
#1194 opened Jan 23, 2025 by abhinavgoel95 • Draft
4 tasks done
@gobbleturk @yangyuwei
3
[WIP] Add GPU MoE e2e test script for XLML testing
#1191 opened Jan 23, 2025 by michelle-yooh Loading…
4 tasks done
3
Initialize distributed IDs on CPU
#1188 opened Jan 22, 2025 by MattIrv Loading…
4 tasks done
@xuefgu
1
Refactor tpu e2e test files
#1186 opened Jan 22, 2025 by khatwanimohit Loading…
4 tasks done
@shralex @gobbleturk @A9isha
14
Simplest maxtext pipeline
#1181 opened Jan 20, 2025 by gobbleturk • Draft
4 tasks
Adding back building images with mode=stable pull ready
#1173 opened Jan 16, 2025 by parambole Loading…
4 tasks done
1
Allow nsys profile saved to local directory
#1160 opened Jan 10, 2025 by hengtaoguo • Draft
3 of 4 tasks
1
[draft] equivalent mixtral mlperf data pipeline
#1157 opened Jan 10, 2025 by ZhiyuLi-goog Loading…
5 tasks done
@suexu1025 @RissyRan
Incorporate Orbax emergency replicator checkpoint manager pull ready
#1153 opened Jan 8, 2025 by xuefgu Loading…
4 tasks done
Anisha seq parallel
#1147 opened Jan 6, 2025 by A9isha • Draft
4 tasks
Fix typo pull ready
#1141 opened Jan 4, 2025 by kislaykishore Loading…
ProTip! What’s not been updated in a month: updated:<2025-01-01.