Independent project. Not a U.S. government website.

Support our project (opens Buy Me a Coffee)
USASI
Generated sky blue circuit-trace artwork for google/paxml, labeled USASI · AI on GitHub, with Organization · Python · Apache-2.0.
Rank #8,919 560 starsPython

paxml

by google (organization)

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. (description from the project’s GitHub page)

Repository facts

Stars
560
Forks
77
Open issues and pull requests
30
Primary language
Python
License
Apache License 2.0 (Apache-2.0)
Owner
google · Organization
Created
Jun 14, 2022
Last push
Oct 15, 2026

Topics

Rank #8,919 of 9,507 in AI on GitHub, by stars. Data from the GitHub API, fetched Oct 18, 2026; star counts change daily. Stars measure popularity, not quality or safety, and a listing is not an endorsement. How this list is made

Add this badge to your README

Ranked #8,919 in AI on GitHubShows this repository’s rank in AI on GitHub and links back to this page. Ranks are by stars and change when the list is refreshed. The badge is not an endorsement or a rating. About the badges

Markdown (for a README)

[![Ranked #8,919 in AI on GitHub](https://unitedstatesofamericasuperintelligence.com/badges/github/google--paxml.svg)](https://unitedstatesofamericasuperintelligence.com/github/google--paxml/)

HTML

<a href="https://unitedstatesofamericasuperintelligence.com/github/google--paxml/"><img src="https://unitedstatesofamericasuperintelligence.com/badges/github/google--paxml.svg" alt="Ranked #8,919 in AI on GitHub" height="20"></a>

Projects that share this one’s GitHub topics.

Support Us

Help keep USASI useful.

Optional. No USASI account required. Payment takes place on the linked provider’s website (Buy Me a Coffee).

About supporting this project