Firstly, you are using the np function, but I believe you won’t get graded properly if you don’t use the trax.fastmath’s numpy, which is fastnp.
Secondly, you are ignoring the code comment which is telling you the variable to use: # the second mask to extract elements in the negative_zero_on_duplicate matrix that are larger than the elements in the diagonal.