エントリーの編集
エントリーの編集は全ユーザーに共通の機能です。
必ずガイドラインを一読の上ご利用ください。
エントリーの編集は全ユーザーに共通の機能です。
必ずガイドラインを一読の上ご利用ください。
注目コメント算出アルゴリズムの一部にLINEヤフー株式会社の「建設的コメント順位付けモデルAPI」を使用しています
Preface このシリーズは、JAXのドキュメントを読みながら学習した際に、ノート代わりに記事に残したもの... Preface このシリーズは、JAXのドキュメントを読みながら学習した際に、ノート代わりに記事に残したものである。JAXを理解する際の要所は、XLAやjust-in-time、Asynchronous dispatchなど多岐にわたるが、この記事ではまず第一弾として、Numpy-likeなAPIで利用でき、自動微分にも対応するJAXのアクセラレートされたNumpyとしての機能(便宜的にJAX numpyと呼ぶ。)についてまとめる。 jax.readthedocs.io 目次 Preface What’s JAX? JAX vs. Numpy JAX API layering Getting Start with JAX numpy JAX array JAX first transformation: grad Value and Grad Auxiliary data First JA