Python + Numpyプログラムの合成可能な変換
- ソフト名 : JAX
- OS : Linux,Windows
- 言語 : Python
- 開発者 : HOME
JAX 詳細説明
その最新バージョンのAutoGradを使用すると、JAXは自動的にネイティブのPythonとNumpy関数を区別できます。それはループ、枝、再帰、および閉鎖を介して区別することができ、そしてそれはデリバティブの誘導体の誘導体を取り得ることができる。勾配微分とフォワードモードの微分を介して逆モードの微分（A.K.A.BACPRAPAGAGATION）をサポートし、2つは任意の順序に任意に構成できます。新機能JAXはXLAを使用してGPUとTPU上のNumpyプログラムをコンパイルして実行します。コンパイルはデフォルトでフードの下で起こり、ライブラリ呼び出しはまさに単にコンパイルされて実行されます。しかしJAXでは、単に独自のPython関数をコンパイルすることもできます。コンパイルと自動差別化は任意に構成できます。そのため、洗練されたアルゴリズムを表現し、Pythonを残さずに最大のパフォーマンスを得ることができます。