JAXという自動微分ライブラリが流行りそうな機運があるので遊んでみます。 github.com インストール READMEに書いてある通りにやりました。 pip install --upgrade pip pip install --upgrade jax jaxlib 基本的な使い方 The Autodiff Cookbook — JAX docume…