Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
101 Open 870 Closed
101 Open 870 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

update setup_gcsfuse for better perf
#1070 opened Nov 27, 2024 by aireenmei Loading…
4 tasks done
Add Pallas GPU decode attention in Maxtext inference
#1066 opened Nov 26, 2024 by tohaowu Loading…
2 of 4 tasks
8
Quantize megablox
#1062 opened Nov 25, 2024 by lenscloth Loading…
1
Make tolerance configurable
#1058 opened Nov 21, 2024 by Doris26 Loading…
1
[DO NOT MERGE] verify fix
#1045 opened Nov 16, 2024 by RissyRan • Draft
[Do Not Merge] v6e scale testing goodput
#1043 opened Nov 15, 2024 by SujeethJinesh • Draft
Add Llama2-70b sparsecore collective model to trillium configs
#1042 opened Nov 15, 2024 by Obliviour Loading…
Enable pathways workloads for v6e benchmarks
#1040 opened Nov 15, 2024 by sadikneipp Loading…
3
Remove aqt einsum for dropping pull ready
#1036 opened Nov 14, 2024 by mailvijayasingh Loading…
@vipannalla
13
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
[MoE] fix typo and add normalization for top_k_weights
#1026 opened Nov 12, 2024 by ZhiyuLi-goog Loading…
@gobbleturk
8
Update maxtext_xpk_runner.py to show how to use vertex ai integrations
#1020 opened Nov 8, 2024 by Obliviour • Draft
[WIP] Support DiLoCo training
#1018 opened Nov 7, 2024 by jonb377 • Draft
3
[DO NOT SUBMIT] Rotate via shmap + ppermute to test new mask idea
#1014 opened Nov 5, 2024 by gobbleturk • Draft
[WIP] Allow PP + megablox
#1009 opened Nov 2, 2024 by gobbleturk Loading…
1
Create Sequence_Packing.md
#1007 opened Nov 1, 2024 by mattdonati Loading…
Fix multihost_runner.py kill script for v6e TPUs pull ready
#1003 opened Oct 31, 2024 by ianmcampbell Loading…
2
Update max_utils.py to run with one process per node
#996 opened Oct 30, 2024 by abhinavgoel95 • Draft
add Gemma2 model configs for trillium regression testing
#994 opened Oct 30, 2024 by wenxindongwork Loading…
Add basic documentation pages
#988 opened Oct 25, 2024 by melissawm Loading…
16
Add qkv_proj to the saved name list
#965 opened Oct 11, 2024 by zhenying-liu • Draft
1
Add memory stats
#964 opened Oct 11, 2024 by zhenying-liu Loading…
1
ProTip! What’s not been updated in a month: updated:<2024-10-29.