Skip to content

Pull requests: google/maxtext

New pull request New
Clear current search query, filters, and sorts
61 Open 600 Closed
61 Open 600 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

Code pieces must be re-written to transition MaxText to Support PyTorch/XLA
#86 opened Jul 14, 2023 by miladm • Draft updated Jul 18, 2023
@miladm
18
Add support for curl command
#102 opened Jul 31, 2023 by tonyjohnchen Loading… updated Aug 28, 2023
8
[WIP] Llama2 Integration
#135 opened Sep 7, 2023 by reshinthadithyan Loading… updated Sep 7, 2023
1 of 3 tasks
1
Add custom remat poicy
#88 opened Jul 18, 2023 by raymondzouu • Draft updated Sep 13, 2023
1
Add main and replace argparse with flags to run sharding on g3
#177 opened Sep 25, 2023 by michelle-yooh Loading… updated Sep 25, 2023
1
Load data on only slice 0
#179 opened Sep 26, 2023 by gobbleturk Loading… updated Sep 26, 2023
Read checkpoint steps from GCS only once, broadcast to rest
#191 opened Sep 29, 2023 by gobbleturk Loading… updated Sep 30, 2023
1
Setup GCSFuse
#173 opened Sep 22, 2023 by khatwanimohit Loading… updated Oct 11, 2023
profiler to start only for the last step
#198 opened Oct 6, 2023 by ultrons Loading… updated Oct 23, 2023
Simplify the mulithost data loading code
#258 opened Nov 13, 2023 by ZacCranko • Draft updated Nov 13, 2023
fixed integer check bug
#255 opened Nov 9, 2023 by grad0s Loading… updated Nov 14, 2023
2
test
#288 opened Dec 12, 2023 by michelle-yooh • Draft updated Dec 12, 2023
Custom metrics emitting
#170 opened Sep 22, 2023 by priyanka-ganesha • Draft updated Dec 16, 2023
24
Increase --shm-size for docker run on GPU pull ready
#298 opened Dec 21, 2023 by michelle-yooh Loading… updated Dec 21, 2023
Fix multihost_job cloud logging in case of maintenance event + recovery
#313 opened Jan 5, 2024 by gobbleturk Loading… updated Jan 5, 2024
@xuefgu
Debug Strange GPU Bug
#340 opened Jan 13, 2024 by rwitten Loading… updated Jan 13, 2024
[Don't Merge] Multihost Decode Monkey Patch
#361 opened Jan 20, 2024 by rwitten Loading… updated Jan 20, 2024
1
[WIP] CPU Unit Tests
#365 opened Jan 22, 2024 by priyanka-ganesha Loading… updated Jan 24, 2024
[Draft] mixtral inference
#369 opened Jan 23, 2024 by WendyShang Loading… updated Jan 25, 2024
2
[DO NOT MERGE] Goodput prototype v1
#337 opened Jan 13, 2024 by dipannita08 Loading… updated Jan 29, 2024
8
[NOT FOR MERGE] Test no metrics
#384 opened Jan 30, 2024 by rwitten Loading… updated Jan 30, 2024
[DO NOT MERGE] Profile Data Loading
#398 opened Feb 6, 2024 by anfals • Draft updated Feb 6, 2024
Example inference workload
#402 opened Feb 7, 2024 by rwitten Loading… updated Feb 7, 2024
added support for cudnn flash attention
#408 opened Feb 10, 2024 by kocchop • Draft updated Feb 17, 2024
14
Anisha ckpt perf
#443 opened Feb 22, 2024 by A9isha • Draft updated Feb 26, 2024
ProTip! no:milestone will show everything without a milestone.