Just to note that this Is in the scope of Recommenders Addons:

  
      

      github.com
  

  
    

GitHub - tensorflow/recommenders-addons: Additional utils and helpers to extend... - Scope 4


  Additional utils and helpers to extend TensorFlow when build recommendation systems, contributed and maintained by SIG Recommenders. - GitHub - tensorflow/recommenders-addons: Additional utils and ...