AIbase
Product LibraryTool Navigation

paxml

Public

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.

Creat2022-06-15T00:04:48
Update2025-03-24T17:50:26
485
Stars
0
Stars Increase

Related projects