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,712 (last one 6 days ago)

Stargazers 475 (2 this week)

Watchers 15 (0 this week)

Forks 68

License apache-2.0

Ranking

RepositoryStats indexes 608,301 repositories, of these google/paxml is ranked #96,406 (84th percentile) for total stargazers, and #147,327 for total watchers. Github reports the primary language for this repository as Python, for repositories using this language it is ranked #16,135/122,654.

google/paxml is also tagged with popular topics, for these it's ranked: llm (#761/3093), gpt (#292/1168), large-language-models (#287/1129)

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 15 open issues and 6 closed issues.

There have been 11 releases, the latest one was published on 2024-04-09 (9 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,712 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-01-28 @ 01:28am, id: 503429749 / R_kgDOHgG6dQ