/micro-grad

A scalar-valued autodiff engine and a neural net library on top of it with PyTorch-like API.

Primary LanguageJupyter NotebookMIT LicenseMIT

Stargazers