paxml

Pax 是一個基於 Jax 的機器學習框架，用於訓練大規模模型。Pax 允許進階且完全可設定的實驗與平行化

paxml 是什麼？

Pax 是一個基於 Jax 的機器學習框架，用於訓練大規模模型。Pax 允許進階且完全可設定的實驗與平行化，並已展現業界領先的模型 FLOP 利用率。