#304 raised the concern that there is some divergence between pytorch and numpy which would prevent ipie's arraylib (some routing for agnostic arraylibrary usage (cupy and numpy)) being used. I don't think it's too much work to just add some custom methods which appropriately dispatch things (such as matrix exp etc), which would reduce a lot of code duplication.