Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
108 Open 948 Closed
108 Open 948 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

Add prefill packing for offline inference
#1162 opened Jan 13, 2025 by sixiang-google Loading…
4 tasks done
Add MoE block unit tests
#1161 opened Jan 11, 2025 by RissyRan Loading…
4 tasks done
@gobbleturk
Allow nsys profile saved to local directory
#1160 opened Jan 10, 2025 by hengtaoguo • Draft
3 of 4 tasks
1
Add environment variable to change Jetstream version
#1159 opened Jan 10, 2025 by vivianrwu Loading…
4 tasks done
[draft] equivalent mixtral mlperf data pipeline
#1157 opened Jan 10, 2025 by ZhiyuLi-goog Loading…
5 tasks done
@suexu1025 @RissyRan
Changes required for gcs-tess ramdisk checkpointing
#1154 opened Jan 8, 2025 by MattIrv • Draft
4 tasks
Incorporate Orbax emergency replicator checkpoint manager pull ready
#1153 opened Jan 8, 2025 by xuefgu Loading…
4 tasks done
Feature/gcp workload monitoring
#1150 opened Jan 7, 2025 by jcyang43 Loading…
4 tasks done
@dipannita08
16
Anisha seq parallel
#1147 opened Jan 6, 2025 by A9isha • Draft
4 tasks
Fix typo
#1141 opened Jan 4, 2025 by kislaykishore Loading…
Raymondzou context parallelism pull ready
#1135 opened Jan 2, 2025 by A9isha • Draft
4 tasks
[DO NOT SUBMIT] Debugging conv w/ offloading pull ready
#1134 opened Jan 2, 2025 by gobbleturk • Draft
4 tasks
Add Context Parallelism support to cudnn Flash Attention
#1133 opened Jan 1, 2025 by kocchop Loading…
4 tasks done
4
Temporary solution to remove the unnecessary ar cache during prefill
#1115 opened Dec 20, 2024 by zhihaoshan-google • Draft
4 tasks done
Get all tests to pass locally with no special configuration
#1108 opened Dec 19, 2024 by SamuelMarks Loading…
4 tasks done
1
2
Anisha ckpt2hf1
#1106 opened Dec 19, 2024 by A9isha • Draft
4 tasks
Add a Ray trainer for MaxText
#1098 opened Dec 13, 2024 by richardsliu Loading…
Add llama-405b configuration for v5p
#1095 opened Dec 10, 2024 by suexu1025 Loading…
4 tasks done
1
Add mixtral 8x7b config for gpu
#1090 opened Dec 9, 2024 by michelle-yooh Loading…
4 tasks done
@michelle-yooh @gobbleturk
1
Enable c4/en, c4/multilingual-en dataset and convergence configs for llama convergence tests
#1080 opened Dec 3, 2024 by suexu1025 • Draft
4 tasks done
1
Add Pallas GPU decode attention in Maxtext inference
#1066 opened Nov 26, 2024 by tohaowu Loading…
4 tasks done
9
[DO NOT MERGE] verify fix
#1045 opened Nov 16, 2024 by RissyRan • Draft
Add Llama2-70b sparsecore collective model to trillium configs
#1042 opened Nov 15, 2024 by Obliviour Loading…
4 tasks done
Enable pathways workloads for v6e benchmarks
#1040 opened Nov 15, 2024 by sadikneipp Loading…
3
Enable orbax cloud logger by default.
#1034 opened Nov 13, 2024 by abhinavclemson Loading…
3
ProTip! What’s not been updated in a month: updated:<2024-12-12.