JAX学習記録④ーHow Autodiff is made?

Preface この記事ではJAX学習記録シリーズ第四弾として、JAXの自動微分がどのように実現されているか、詳細を見てみる。正直なところ、特に数学的な説明に関して理解しきれていないところが多く、基本的には原文に忠実に訳しており、自分の理解による説明はあまり加えられていない。もし修正が必要な個所があったり、もっ…