Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
139 Open 1,058 Closed
139 Open 1,058 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 Disruption Manager & Suspend/Resume Test Script
#1312 opened Feb 26, 2025 by SujeethJinesh • Draft
4 tasks
Update models.py to fix num_layers issue for scan_layer=False in Deep…
#1311 opened Feb 25, 2025 by deval281shah Loading…
2 tasks
4
Poc elastic training
#1310 opened Feb 25, 2025 by lukebaumann • Draft
4 tasks
[gitignore] Add '.idea' dir in .gitignore pull ready
#1309 opened Feb 25, 2025 by wyzhang Loading…
4 tasks done
Sujinesh/add checkpointing to benchmarks
#1306 opened Feb 24, 2025 by SujeethJinesh • Draft
4 tasks
integrate gpu pallas flash attention . Reduce prefill time for llama70b pull ready
#1305 opened Feb 24, 2025 by jwyang-google Loading…
4 tasks done
@tohaowu
Add flops calculation for DeepSeek pull ready
#1304 opened Feb 24, 2025 by RissyRan Loading…
4 tasks done
@gagika
7
LLama 405B Tweaks
#1298 opened Feb 22, 2025 by anfals • Draft
4
Update PULL_REQUEST_TEMPLATE.md
#1297 opened Feb 22, 2025 by shralex Loading…
4 tasks done
1
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
39
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
2
BigQuery Support
#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
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.