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_atNov. 12, 2024, 4:14 p.m.
pushed_atNov. 12, 2024, 4:14 p.m.
size4,250
stargazers_count457
watchers_count15
forks_count68
open_issues30
languagePython
awesome_list

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