Extend DistributedLossWrapper to use with ref_emb and ref_labels

[Bomme]

Hi,
first of all thanks for this amazing library!
Do you have any plans for extending the functionality of DistributedLossWrapper for it to be used with ref_emb and ref_labels?

[KevinMusgrave]

Yes, I just need to review and merge this code: #503

[Bomme]

Thanks for the quick reply! I missed that PR! (I only checked issues and discussions)

[KevinMusgrave]

This feature is now available in v1.6.0

[Bomme]

Many thanks!