Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
107 Open 919 Closed
107 Open 919 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

[Do Not Merge] gpt3 test in parallel tests
#1130 opened Dec 27, 2024 by ZhiyuLi-goog Loading…
4 tasks done
Add option to skip initializing the jax distributed system
#1125 opened Dec 23, 2024 by gobbleturk Loading…
4 tasks done
2
Fix orbax to hf converter for Llama3.1-8B
#1123 opened Dec 23, 2024 by khatwanimohit Loading…
4 tasks done
@A9isha
4
Add option to dump hlo
#1121 opened Dec 23, 2024 by gobbleturk Loading…
4 tasks done
@hengtaoguo @raymondzouu
Incorporate Orbax emergency replicator checkpoint manager
#1117 opened Dec 21, 2024 by xuefgu • Draft
4 tasks done
Temporary solution to remove the unnecessary ar cache during prefill
#1115 opened Dec 20, 2024 by zhihaoshan-google • Draft
4 tasks done
Refactor and parallelize MaxText test runner
#1113 opened Dec 20, 2024 by shralex Loading…
4 tasks done
19
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
Fix typo in trillium llama2-70b script
#1105 opened Dec 19, 2024 by wenxindongwork Loading…
4 tasks done
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 Pathways Support to Benchmark Runner
#1094 opened Dec 10, 2024 by SujeethJinesh Loading…
24
Add mixtral 8x7b config for gpu
#1090 opened Dec 9, 2024 by michelle-yooh Loading…
4 tasks done
@michelle-yooh @gobbleturk
1
Enable c4/en, c4/multilingual-en dataset and convergence configs for llama convergence tests
#1080 opened Dec 3, 2024 by suexu1025 • Draft
4 tasks done
1
[DO NOT MERGE] Pathways llama 405b v5p config
#1077 opened Dec 3, 2024 by SujeethJinesh • Draft
Add Pallas GPU decode attention in Maxtext inference
#1066 opened Nov 26, 2024 by tohaowu Loading…
4 tasks done
9
Quantize megablox
#1062 opened Nov 25, 2024 by lenscloth Loading…
4 tasks done
40
[DO NOT MERGE] verify fix
#1045 opened Nov 16, 2024 by RissyRan • Draft
Add Llama2-70b sparsecore collective model to trillium configs
#1042 opened Nov 15, 2024 by Obliviour Loading…
4 tasks done
Enable pathways workloads for v6e benchmarks
#1040 opened Nov 15, 2024 by sadikneipp Loading…
3
Enable orbax cloud logger by default.
#1034 opened Nov 13, 2024 by abhinavclemson Loading…
3
ray fault tolerance
#1032 opened Nov 13, 2024 by keshavb96 • Draft
[DO NOT MERGE] Ranran hide a2a
#1029 opened Nov 12, 2024 by RissyRan • Draft
5
Support for safetensor checkpoints
#1028 opened Nov 12, 2024 by richjames0 • Draft
1
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.