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

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