@pgree experienced a similar issue which we tracked down to instability in one of our dependencies, which was fixed but not yet released by them: Nans in complex eigendecomposition
In his case the issue seemed to be tied to very small but not quite zero values - maybe something similar is happening here, which is why the equality test is failing? (print()
will do some rounding)