Abstract: Aspects of the present disclosure provide techniques for training a machine learning model. Embodiments include providing features of a plurality of content items as inputs to an embedding model and receiving embeddings of the plurality of content items as outputs from the embedding model. Embodiments include receiving a data set comprising features of a plurality of users associated with content items of the plurality of content items that correspond to the plurality of users. Embodiments include generating a training data set for a machine learning model, wherein the training data set comprises the features of the plurality of users associated with respective labels indicating which respective embeddings of the embeddings correspond to each respective user of the plurality of users. Embodiments include training the machine learning model, using the training data set, to output corresponding embeddings of relevant content items for users based on features of the users.