Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
102 Open 859 Closed
102 Open 859 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

Adding a PR description template to MaxText
#1060 opened Nov 24, 2024 by shralex Loading…
4 tasks done
2
Ads parallel MLPerf accuracy eval. pull ready
#1059 opened Nov 22, 2024 by patemotter Loading…
@vipannalla @singh-mitali
4
Make tolerance configurable
#1058 opened Nov 21, 2024 by Doris26 Loading…
1
Merge the disaggregation prototyping code pull ready
#1049 opened Nov 19, 2024 by zhihaoshan-google Loading…
@singh-mitali
15
Add checkpoint topology discovery for the replicator service
#1048 opened Nov 18, 2024 by xuefgu Loading…
[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
[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
ProTip! Filter pull requests by the default branch with base:main.