Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
136 Open 1,043 Closed
136 Open 1,043 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 DeepSeek V2 config
#1293 opened Feb 21, 2025 by RissyRan Loading…
4 tasks done
@gagika
2
Add gpu pallas flash attention for inference. Collobrate with tohaowu.
#1292 opened Feb 20, 2025 by jwyang-google Loading…
4 tasks
Enables the huggingface checkpoint conversion to MaxText orbax. pull ready
#1291 opened Feb 20, 2025 by wang2yn84 Loading…
4 tasks done
@richjames0 @RissyRan @vipannalla
25
fix end of batch cache slicing
#1289 opened Feb 20, 2025 by sixiang-google Loading…
4 tasks done
@vipannalla @singh-mitali
2
Restructure Pyconfig pull ready
#1285 opened Feb 19, 2025 by A9isha Loading…
4 tasks done
@shralex
10
Add llama3.3-70b-instruct model configs, tests and HF golden logits
#1282 opened Feb 18, 2025 by hengtaoguo Loading…
4 tasks done
create separate emergency ckpt directories for each process
#1281 opened Feb 18, 2025 by keshavb96 Loading…
@xuefgu
Prefix Caching with HBM and latency test
#1278 opened Feb 17, 2025 by yuyanpeng-google Loading…
4 tasks done
13
Link to maxtext_xpk_runner in readme for getting started with XPK pull ready
#1275 opened Feb 14, 2025 by Obliviour Loading…
4 tasks done
fix: Pin orbax-checkpoint to v0.10.3 to resolve dependency error (#1273)
#1274 opened Feb 14, 2025 by shota-inoue-lts Loading…
@A9isha
3
[WIP]chunked prefill
#1272 opened Feb 14, 2025 by mailvijayasingh Loading…
4 tasks done
7
Added support of training with NANOO fp8 GEMM on AMD MI300/MI325 GPUs.
#1262 opened Feb 11, 2025 by wenchenvincent Loading…
4 tasks done
1
bq draft
#1250 opened Feb 7, 2025 by Obliviour Loading…
4 tasks done
13
Use causal mask type for training or prefill with fixed seqlen when
#1248 opened Feb 7, 2025 by wenchenvincent • Draft
3 of 4 tasks
Enable mlperf to use mixtral-v1 tokenized dataset to avoid dropping in total_weights when splitting data across multiple host
#1246 opened Feb 7, 2025 by aireenmei Loading…
4 tasks done
Upload HLOs
#1244 opened Feb 6, 2025 by Obliviour • Draft
4 tasks
1
POC for GPU e2e feature testing
#1243 opened Feb 6, 2025 by sfvaroglu • Draft
Gpu inference
#1240 opened Feb 5, 2025 by jwyang-google Loading…
4 tasks
[wip-prototype]chunked prefill on 1k prompt len
#1237 opened Feb 4, 2025 by mailvijayasingh Loading…
4 tasks
Add a basic sampling test
#1235 opened Feb 4, 2025 by liurupeng Loading…
4 tasks done
2
MaxText Trillium Benchmarking
#1224 opened Jan 31, 2025 by anfals • Draft
4 tasks
[NVIDIA] Support FP8 quantization for MOE layers
#1221 opened Jan 31, 2025 by kaixih Loading…
4 tasks done
23
Add Pathways Recipe Support for Scale Testing
#1220 opened Jan 31, 2025 by SujeethJinesh Loading…
4 tasks done
[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
ProTip! Type g i on any issue or pull request to go back to the issue listing page.