Currently all computations are done using numpy only. It would be good to provide some flexibility, so users can use torch, cupy, etc.