Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.
Run machine learning experiments and train large models using Jax on cloud computing resources.
Claim its page: indexed whatever its rank, translated into six languages, and enriched with what you write yourself.
Get an email alert on its next release or when it starts trending — never miss the moment.
Free · no card · unsubscribe anytimePax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.
paxml has 558 stars on GitHub. It has been forked 72 times. paxml is written mainly in Python. It has been in active development since 2022. paxml is available under the Apache-2.0 license. Its main topics are c4, gpt, jax, large-language-models.
Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.
paxml is an open-source project. It is released under the Apache-2.0 license.
Yes. paxml is free and open source — you can use, modify and self-host it.
paxml is available under the Apache-2.0 license.
paxml is written mainly in Python.
Add this live badge to your README — your GitHub stars and directory rank, refreshed daily.
[](https://olud.ai/project/google-paxml.html)