[FEATURE] Migrate TensorFlow algorithms to PyTorch

Author: miguelgfierroCreated Apr 6, 2026Updated Aug 24, 2026
Labelsenhancement

Description

Curent TF algos:

 Here are the TensorFlow-based models in the repo:

  ┌────────────┬──────────────────────────────────────────────────────────────────────────────┐
  │ Subpackage │                                    Models                                    │
  ├────────────┼──────────────────────────────────────────────────────────────────────────────┤
  │ deeprec    │ XDeepFM, DKN, DKN_Item2Item, LightGCN, GRU, Caser, A2SVD, NextItNet, SLI_REC │
  ├────────────┼──────────────────────────────────────────────────────────────────────────────┤
  │ newsrec    │ NAML, NRMS, NPA, LSTUR                                                       │
  ├────────────┼──────────────────────────────────────────────────────────────────────────────┤
  │ ncf        │ Neural Collaborative Filtering                                               │
  ├────────────┼──────────────────────────────────────────────────────────────────────────────┤
  │ vae        │ MultinomialVAE, StandardVAE                                                  │
  ├────────────┼──────────────────────────────────────────────────────────────────────────────┤
  │ rbm        │ Restricted Boltzmann Machine                                                 │
  └────────────┴──────────────────────────────────────────────────────────────────────────────┘

  That's 5 subpackages with ~17 model implementations total. The deeprec and newsrec packages are the largest, each with their own base class hierarchy. Most use
  TF 1.x compatibility mode (tf.compat.v1), while newsrec uses Keras.

Expected behavior with the suggested feature

Working in PyTorch

Willingness to contribute

  • Yes, I can contribute for this issue independently.
  • Yes, I can contribute for this issue with guidance from Recommenders community.
  • No, I cannot contribute at this time.

Other Comments

Source: recommenders-team/recommenders