Also the full traces here:
0.4.0 https://ufile.io/73ief 2
1.1.0 https://ufile.io/t0b1x 3
Among the things that stand out is that the backwards pass of 1.1.0 seems to be made up of almost entirely blocks of TrtrsBackward, followed by SelectBackward and add, versus 0.4.0 which is using MulBackward, PowBackwards, etc. Does this point towards a LAPACK issue?