Autograd engine + tensor library + wgpu backend
by Michal Hornický
#25 in #autograd
70KB 2K SLoC
Rust Autograd + tensor lib + gpu backend.
~21–51MB ~817K SLoC