kron-torch 0.3.1

What's Changed

• Improve merge dims option and default to True. Now merging dims finds the most square matrix to reshape grad tensors into.