(Full PyTorch Code Reference: https://github.com/openai/CLIP)