See https://github.com/oxpig/FlashABB/pull/1 This will be easy if the MPS kernel allows mismatched dimensions. Otherwise it would be worth profiling the performance hit of padding the dimensions.
See #1
This will be easy if the MPS kernel allows mismatched dimensions. Otherwise it would be worth profiling the performance hit of padding the dimensions.