Loreon
Labs
Platform
Docs
Home
Ecosystems
Other
maxtext-custom
A simple, performant and scalable Jax LLM!
Other
Emerging
GitHub
Stars
—
Forks
—
Contributors
8
Last push
27mo ago
Recent commits
Latest commits.
Revert "Add llama2 configs"
5deb01a
michelle-yooh
27mo ago
Add llama2 configs
b4c4139
michelle-yooh
27mo ago
Merge pull request #573 from google:yooh/minimal_flash
264c823
maxtext authors
27mo ago
Merge pull request #568 from google:anfals/params_streamline_redo
de49d83
maxtext authors
27mo ago
Merge pull request #556 from google:ss-sharding-utest
eb4a173
maxtext authors
27mo ago
add validation for shardings
6820b5b
ssusie
27mo ago
Add minimal remat policy for flash attention
012e0a4
michelle-yooh
27mo ago
Removes unnecessary if block.
5575702
Yunlong Liu
27mo ago
Top contributors
Builders behind this project.
gobbleturk
152 commits
raymondzouu
85 commits
khatwanimohit
81 commits
michelle-yooh
46 commits
tonyjohnchen
45 commits
Obliviour
40 commits
priyanka-ganesha
31 commits
A9isha
23 commits
