Skip to content

Pull requests: google/maxtext

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

Code pieces must be re-written to transition MaxText to Support PyTorch/XLA
#86 opened Jul 14, 2023 by miladm • Draft
@miladm
18
Add custom remat poicy
#88 opened Jul 18, 2023 by raymondzouu • Draft
1
Custom metrics emitting
#170 opened Sep 22, 2023 by priyanka-ganesha • Draft
24
Setup GCSFuse
#173 opened Sep 22, 2023 by khatwanimohit Loading…
Load data on only slice 0
#179 opened Sep 26, 2023 by gobbleturk Loading…
Read checkpoint steps from GCS only once, broadcast to rest
#191 opened Sep 29, 2023 by gobbleturk Loading…
1
profiler to start only for the last step
#198 opened Oct 6, 2023 by ultrons Loading…
Simplify the mulithost data loading code
#258 opened Nov 13, 2023 by ZacCranko • Draft
test
#288 opened Dec 12, 2023 by michelle-yooh • Draft
Increase --shm-size for docker run on GPU pull ready
#298 opened Dec 21, 2023 by michelle-yooh Loading…
Fix multihost_job cloud logging in case of maintenance event + recovery
#313 opened Jan 5, 2024 by gobbleturk Loading…
@xuefgu
[DO NOT MERGE] Goodput prototype v1
#337 opened Jan 13, 2024 by dipannita08 Loading…
8
Debug Strange GPU Bug
#340 opened Jan 13, 2024 by rwitten Loading…
Collect metrics for GCS tessellation
#348 opened Jan 17, 2024 by llukai • Draft
2
CPU Inference Configs
#364 opened Jan 22, 2024 by priyanka-ganesha Loading…
1
[WIP] CPU Unit Tests
#365 opened Jan 22, 2024 by priyanka-ganesha Loading…
[NOT FOR MERGE] Test no metrics
#384 opened Jan 30, 2024 by rwitten Loading…
[DO NOT MERGE] Profile Data Loading
#398 opened Feb 6, 2024 by anfals • Draft
Example inference workload
#402 opened Feb 7, 2024 by rwitten Loading…
Anisha ckpt perf
#443 opened Feb 22, 2024 by A9isha • Draft
test runner push permission
#452 opened Feb 26, 2024 by gobbleturk Loading…
Ragged attention and test, not actively used.
#453 opened Feb 26, 2024 by patemotter Loading…
1
Docker Build Dependency Image
#454 opened Feb 27, 2024 by gobbleturk Loading…
Anisha jsonl tfrecord
#456 opened Feb 27, 2024 by A9isha • Draft
Mattdavidow build debug
#458 opened Feb 27, 2024 by gobbleturk Loading…
ProTip! Mix and match filters to narrow down what you’re looking for.