Home Projects paxml
paxml
Python

paxml

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.

by google · GitHub
Stars
Forks
License
Created
Last commit
Category
Language
c4gptjaxApache-2.0Python
View on GitHub
In plain words

Run machine learning experiments and train large models using Jax on cloud computing resources.

You maintain this project?

Claim its page: indexed whatever its rank, translated into six languages, and enriched with what you write yourself.

Claim this page →
paxml — GitHub preview card
📈 Star history
559555
2026-07-202026-08-31
📈 Track paxml

Get an email alert on its next release or when it starts trending — never miss the moment.

Free · no card · unsubscribe anytime
Get email alerts →
📄 About

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 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.

Frequently asked questions

What is paxml?

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.

Is paxml open source?

paxml is an open-source project. It is released under the Apache-2.0 license.

Is paxml free?

Yes. paxml is free and open source — you can use, modify and self-host it.

What license does paxml use?

paxml is available under the Apache-2.0 license.

What language is paxml written in?

paxml is written mainly in Python.

🏅 Maintainer of this project?
olud.ai badge — paxml

Add this live badge to your README — your GitHub stars and directory rank, refreshed daily.

[![olud.ai](https://olud.ai/badge.php?tool=google-paxml)](https://olud.ai/project/google-paxml.html)
More badge options →
🧬 Related projects🧬 View the DNA map →