Microjax：2つのクラスと6つの関数でJAXを実現

Andrej KarpathyのMicrogradに触発されたMicrojaxは、わずか2つのクラスと6つの関数でJAXの機能を再現するライブラリです。人気のPyTorchとは異なり、MicrojaxはJAXのより関数的なプログラミングスタイルを採用しています。このチュートリアルは、JAXの前身であるautogradに関するMatthew J Johnsonの優れた2017年のプレゼンテーションを参考に、簡素化してノートブックとしてパッケージ化しています。