"""Distribution-matching losses used by distribution matching."""