Databricks Interview Question

Using pytorch, write a code to train a retrieval system using contrastive loss