code for training & evaluating Contextual Document Embedding models - View it on GitHub