Have you already explored TensorFlow Recommenders (TFRS)?

  
      

      blog.tensorflow.org
  

  
    

Introducing TensorFlow Recommenders 15

  Introducing TensorFlow Recommenders, a library for building flexible and powerful recommender models.