Matrix inversion fails on GPU (google Colab)

2

Does the error happen during the inverse or during the matmul in the print?