Skip to content

Pull requests: google/maxtext

New pull request New
58 Open 556 Closed
58 Open 556 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
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

Pipeline parallelism
#677 opened May 31, 2024 by gobbleturk Loading…
@jonb377 @khatwanimohit
Add MoE e2e test on GPU
#673 opened May 28, 2024 by michelle-yooh • Draft
save and load quantized checkpoint
#671 opened May 24, 2024 by jwyang-google Loading…
5
Add KV Cache Layout Control and Attention Tests pull ready
#667 opened May 23, 2024 by morgandu Loading…
5 of 6 tasks
@yeandy @gobbleturk
8
Sweep across KV cache layouts
#662 opened May 21, 2024 by yeandy • Draft
@patemotter @gobbleturk
23
Llama3-8b 🦙 pull ready
#653 opened May 16, 2024 by khatwanimohit Loading…
29
[Not for Merge] [POC] Goodput async monitoring and upload to Tensorboard POC
#648 opened May 14, 2024 by dipannita08 Loading…
1 task done
2
Add fully qualified path to doclinks.
#638 opened May 2, 2024 by AngryBrock Loading…
Update constraints to the latest stable
#626 opened Apr 29, 2024 by chajath • Draft
[WIP] add debug functionality for per chip sizes and bytes
#625 opened Apr 26, 2024 by morgandu • Draft
20
docs: update Run_MaxText_via_multihost_runner.md
#620 opened Apr 25, 2024 by eltociear Loading…
1
Correct the Run_Gemma.md path in README.md
#617 opened Apr 23, 2024 by hengtaoguo Loading…
Revert "Mark nvidia devtools repo as trusted"
#613 opened Apr 22, 2024 by chajath Loading…
Update First_run.md - Fixed broken links path pull ready
#599 opened Apr 17, 2024 by shivajid Loading…
Reviewed MaxText README.md pull ready
#590 opened Apr 11, 2024 by mikegre-google Loading…
11
[WIP] add batch inference
#584 opened Apr 10, 2024 by morgandu • Draft
Convert Orbax ckpt to HuggingFace
#581 opened Apr 9, 2024 by A9isha Loading…
@gobbleturk
5
fix: typo
#566 opened Mar 28, 2024 by emergenz Loading…
1
Add docker support to maxtext base image
#564 opened Mar 28, 2024 by tonyjohnchen Loading…
2
Fix typos
#559 opened Mar 27, 2024 by BioGeek Loading…
@khatwanimohit
4
Pin nvidia-cudnn-cu12==8.9.7.29
#557 opened Mar 26, 2024 by michelle-yooh Loading…
1
Flip env sensing so we can launch gpu with checkpointing
#540 opened Mar 20, 2024 by chajath Loading…
1
Megatron style TFLOPs Calculation
#537 opened Mar 20, 2024 by abhinavgoel95 Loading…
8
[NOT FOR MERGE] Rwitten host offload demo pull ready
#535 opened Mar 19, 2024 by rwitten Loading…
2
Cut a release branch for GPU runs + flash attention WIP
#525 opened Mar 15, 2024 by chajath • Draft
ProTip! Add no:assignee to see everything that’s not assigned.