Awesome

pytorch autograd solver

DONE

• symeig
• batch_symeig
• generalized_symeig

TODO

• batch_svd
• batch_inv

see torch_autograd_solver_/__init__.py

requirements

• torch 0.4.1 - 0.5.0
• (CUDA optional) https://github.com/ShigekiKarita/pytorch-cusolver