Home Projects paxml
paxml

paxml

by google · GitHub

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.

# c4# gpt# jax# large-language-models
View on GitHub
⭐ Stars
556
🍴 Forks
74
📜 License
Apache-2.0
Commercial use OK
📅 Created
2022
🔄 Last commit
today
🏷️ Category
c4
💻 Language
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
557556
2026-07-202026-07-21
📄 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.

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.

🏅 Maintainer of this project?
OpenSourceAI badge — paxml

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

[![OpenSourceAI](https://opensourceai.tech/badge.php?tool=google-paxml)](https://opensourceai.tech/project/google-paxml.html)
More badge options →
🧬 Shares DNA with🧬 View the DNA map →

Measured from GitHub topics shared by both projects, weighted by how rare each topic is.