Artem

Hi, awesome work! Have you ever had a chance to reproduce results for Proxy-NCA loss? Thx

Adding support for StreamK for GEMM operations (except for group GEMM). As workspace calculations are hard to cap for a particular range of shapes (and max shape calcualation doesn't always...