infoNCELoss. CuBlaze is a CUDA implementation of the InfoNCE Loss (Information Noise-Contrastive Estimation) for self-supervised contrastive learning. This implementation provides full batch processing with native integration of PyTorch autograd and support for multiple optimized implementations.

github.com/GianniMoretti/infoNCELoss

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.