/thxx-py

thxx : pytorch experimental expansion

Primary LanguageC++

thxx : pytorch experimental expansion

DONE

  • backend

    • symeig
    • batch_symeig
    • generalized_symeig
    • complex matmul
    • batch_matinv

  • autograd

    • symeig
    • batch_symeig
    • generalized_symeig

TODO

  • batch_svd
  • batch/complex inv

requirements

  • torch 0.4.1 - 0.5.0

dev

  1. python setup.py install
  2. pytest