keyvalue
id503429749
namepaxml
full_namegoogle/paxml
html_urlhttps://github.com/google/paxml
descriptionPax 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.
created_atJune 14, 2022, 4:04 p.m.
updated_atApril 27, 2024, 3:38 a.m.
pushed_atApril 26, 2024, 5:52 p.m.
size4,359
stargazers_count389
watchers_count15
forks_count54
open_issues24
languagePython
awesome_list

https://github.com/n2cholas/awesome-jax