Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
114 Open 979 Closed
114 Open 979 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

[DRAFT] Add PW/McJAX Benchmark Comparison & Long Running Test Recipes
#1201 opened Jan 26, 2025 by SujeethJinesh • Draft
4 tasks
Add on-device runner and v5e benchmark model configs
#1199 opened Jan 24, 2025 by raymondzouu Loading…
4 tasks done
@Obliviour
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…
2 of 4 tasks
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
Simplest maxtext pipeline
#1181 opened Jan 20, 2025 by gobbleturk • Draft
4 tasks
Remote Python Support for Benchmark Runner
#1174 opened Jan 17, 2025 by SujeethJinesh Loading…
4 tasks done
Adding back building images with mode=stable pull ready
#1173 opened Jan 16, 2025 by parambole Loading…
4 tasks done
1
MMLU Benchmarks
#1163 opened Jan 13, 2025 by gagika Loading…
4 tasks done
@RissyRan @gobbleturk
5
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…
Raymondzou context parallelism pull ready
#1135 opened Jan 2, 2025 by A9isha • Draft
4 tasks
[DO NOT SUBMIT] Debugging conv w/ offloading pull ready
#1134 opened Jan 2, 2025 by gobbleturk • Draft
4 tasks
Add Context Parallelism support to cudnn Flash Attention
#1133 opened Jan 1, 2025 by kocchop Loading…
4 tasks done
4
Temporary solution to remove the unnecessary ar cache during prefill
#1115 opened Dec 20, 2024 by zhihaoshan-google • Draft
4 tasks done
Get all tests to pass locally with no special configuration
#1108 opened Dec 19, 2024 by SamuelMarks Loading…
4 tasks done
1
2
Anisha ckpt2hf1
#1106 opened Dec 19, 2024 by A9isha • Draft
4 tasks
Add a Ray trainer for MaxText
#1098 opened Dec 13, 2024 by richardsliu Loading…
Add llama-405b configuration for v5p
#1095 opened Dec 10, 2024 by suexu1025 Loading…
4 tasks done
1
Add mixtral 8x7b config for gpu
#1090 opened Dec 9, 2024 by michelle-yooh Loading…
4 tasks done
@michelle-yooh @gobbleturk
1
ProTip! Type g i on any issue or pull request to go back to the issue listing page.