Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
93 Open 839 Closed
93 Open 839 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 support for building Maxtext images with nightly version of JSS for GPUs
#1030 opened Nov 12, 2024 by parambole Loading…
2
[DO NOT MERGE] Ranran hide a2a
#1029 opened Nov 12, 2024 by RissyRan • Draft
1
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…
@RissyRan
1
Update maxtext_xpk_runner.py
#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
Fix pipeline rematerialization pull ready
#971 opened Oct 20, 2024 by gobbleturk Loading…
@jonb377
3
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
Bump the pip group across 1 directory with 2 updates dependencies Pull requests that update a dependency file
#961 opened Oct 10, 2024 by dependabot bot Loading…
[DO NOT MERGE] Optimizer host offload pull ready
#960 opened Oct 9, 2024 by khatwanimohit • Draft
Enable llama3.1 405B GPU Training
#956 opened Oct 8, 2024 by wang2yn84 Loading…
Checkpoint tfds data iterator
#954 opened Oct 7, 2024 by mattdonati Loading…
@aireenmei
3
[WIP] DPO
#947 opened Oct 5, 2024 by rdyro Loading…
1
Rebase experimental_proxy on main
#944 opened Oct 4, 2024 by avrittrohwer Loading…
1
Updating PathwaysUtils GitHub organization
#930 opened Oct 1, 2024 by lukebaumann Loading…
Main merge
#907 opened Sep 21, 2024 by kyle-google Loading…
1
[WIP] partial nnx impl
#896 opened Sep 17, 2024 by rdyro • Draft
ProTip! Follow long discussions with comments:>50.