I am not sure if the masked loss has to be used with logits to be true, check again once instructions before the grade masked loss.
next for masked accuracy, refer the below comment to make corrections. Although some of the codes are correct but the way you are implementing might be the issue for the autograder. You do not require to recalled step in two steps for then calculate the masked accuracy. You also need to check if you have used the correct axis arguments