PR2: Update mkmmd loss

[sanaAyrml]

PR Type

[Fix]

Short Description

Clickup Ticket(s): Link(s) if applicable.

This is update version of mkmmd loss to improve is two test power.

Tests Added

Mkmmd test to be fixed if we end up merging these changes.