Issue with Triplet Loss function

Below is the code which I wrote after carefully following the comments and notes although it is asked to implement it differently in Assignment to lab. But I’m unable to move forward with this and unit tests are failing, need assistance in this. Lab Id : xjzgqfax
@arvyzukai @drew_Frances @balaji.ambresh

[code removed - moderator]

fastnp.max is different from fastnp.maximum. Please use this hint to fix your code.

@balaji.ambresh Thank you very much, So I need to understand all the functionalities of trax numpy module.