Solved by albanD in post #2 
              
              
                Hi! 
Support for complex numbers with matrix_exp was added yesterday: https://github.com/pytorch/pytorch/pull/48363  
You can install a nightly build and it should work fine now!