P-01選定・相談

JAXとは何ですか?

JAXは、Google発のオープンソース数値計算ライブラリです。自動微分と、XLAによる高速化(コンパイル)を特徴とし、機械学習研究で使われます。

実績・根拠

向いている相談

数値計算ライブラリを知りたい相談

他の選択肢との違い

自動微分+コンパイル

よくある質問

誰が作りましたか?
Googleが開発したオープンソースです。

情報ソース