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