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

Date Created 2022-06-14 (2 years ago)

Commits 1,716 (last one 4 days ago)

Stargazers 484 (0 this week)

Watchers 13 (0 this week)

Forks 69

License apache-2.0

Ranking

RepositoryStats indexes 632,768 repositories, of these google/paxml is ranked #97,331 (85th percentile) for total stargazers, and #161,683 for total watchers. Github reports the primary language for this repository as Python, for repositories using this language it is ranked #16,535/129,079.

google/paxml is also tagged with popular topics, for these it's ranked: llm (#867/3562), gpt (#302/1234), large-language-models (#312/1230)

Other Information

google/paxml has 15 open pull requests on Github, 40 pull requests have been merged over the lifetime of the repository.

Github issues are enabled, there are 17 open issues and 6 closed issues.

There have been 11 releases, the latest one was published on 2024-04-09 (11 months ago) with the name paxml release 1.4.0.

All Topics

c4 gpt jax llm model-flops parallelism large-language-models

Star History

Github stargazers over time

Watcher History

Github watchers over time, collection started in '23

Recent Commit History

1,716 commits on the default branch (main) since jan '22

Yearly Commits

Commits to the default branch (main) per year

Issue History

Languages

The primary language is Python but there's also others...

updated: 2025-03-25 @ 03:44am, id: 503429749 / R_kgDOHgG6dQ