JAXは自動微分、線形計算を高速に行い、機械学習を実装できるライブラリです。
ログイン・登録して始める chevron_right
JAXの入門
JAXの配列について学ぶ
実行時コンパイルについて学ぶ
自動ベクトル化について学ぶ
勾配について学ぶ
Pytreeについて学ぶ
ニューラルネットワークついて学ぶ