Autograd engine + tensor library + wgpu backend
by Michal Hornický
#23 in #autograd
70KB 2K SLoC
Rust Autograd + tensor lib + gpu backend.
~20–52MB ~846K SLoC