Skip to content

Pull requests: google/maxtext

New pull request New
61 Open 544 Closed
61 Open 544 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

Add KV Cache Layout Control
#667 opened May 23, 2024 by morgandu Loading…
@rwitten @patemotter
@yeandy @vipannalla
5
Add flag for single controller workloads to avoid jax.distributed.initialize()
#666 opened May 22, 2024 by sadikneipp Loading…
Pass metrics client config through to Jetstream
pull ready
#665 opened May 22, 2024 by Bslabe123 Loading…
integrate nsys profiler into MaxText
#663 opened May 21, 2024 by michelle-yooh Loading…
6
Sweep across KV cache layouts
#662 opened May 21, 2024 by yeandy • Draft
6
Add MoE matmul implementation
#659 opened May 21, 2024 by RissyRan • Draft
Llama3-8b 🦙
#653 opened May 16, 2024 by khatwanimohit Loading…
@gobbleturk
17
[Not for Merge] [POC] Goodput async monitoring and upload to Tensorboard POC
#648 opened May 14, 2024 by dipannita08 Loading…
1 task done
2
fix sharding for generate cache in prefill results
pull ready
#641 opened May 7, 2024 by jwyang-google Loading…
12
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
add debug functionality for per chip sizes and bytes
#625 opened Apr 26, 2024 by morgandu Loading…
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 llama2-13b tests
#565 opened Mar 28, 2024 by morgandu Loading…
@rwitten
3
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
ProTip! Filter pull requests by the default branch with base:main.