Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
90 Open 827 Closed
90 Open 827 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 SUBMIT] Rotate via shmap + ppermute to test new mask idea
#1014 opened Nov 5, 2024 by gobbleturk • Draft
fix data input on single CPU host
#1013 opened Nov 4, 2024 by aireenmei Loading…
1
[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…
optimizations for offline mlperf inference
#1006 opened Nov 1, 2024 by sixiang-google Loading…
2
Upgrade TE GPU pin to 1.11.0
#1004 opened Nov 1, 2024 by gobbleturk 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 • Draft
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
Initialize jax distributed when checkpointing is enabled
#895 opened Sep 16, 2024 by jonb377 Loading…
7
Docker prune in all github actions
#890 opened Sep 13, 2024 by khatwanimohit • Draft
1
removing tensorflow_text for aarch64 compatiblity pull ready
#883 opened Sep 12, 2024 by rdyro • Draft
@aireenmei
6
ProTip! Type g i on any issue or pull request to go back to the issue listing page.