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