Code implementation of Enhancing Context Using Contrast NeurIPS Pre-Registration Workshop paper - View it on GitHub