Solved by planet in post #3 
              
              
                I was using earlier versions of Pytorch 1.4 where I could not find torch.linalg.eig. I updated it to newest version 1.9.0 and I able to get identical results comparing with Tensorflow. 
Abhi