joelburget/microjax

A tiny autograd engine with a Jax-like API

语言：Jupyter Notebook · GitHub 星标：76 · Fork：1

在 GitHub 上查看原仓库