handle if even amount of batch sizes across devices

export the sigmoid contrastive learning module

further cleanup the distributed code

quick fix

complete distributed logic for sigmoid contrastive loss

make sure regular contrastive loss supports distributed data parallel…

…. work towards completing the sigmoid contrastive loss

move beartype onto methods

missing import

add sigmoid contrastive loss

contrastive loss needs both directions