NVIDIA Merlin is an open source library providing end-to-end GPU-accelerated recommender systems, from feature engineering and preprocessing to training deep learning models and running inference in production.
Apache License 2.0
758
stars
113
forks
source link
[RMP] Add support for ranking models in PyTorch #1044
Currently the block-API is T4Rec is using a similar design to Keras to allow for modules that lazily initialize their variables. We would like to deprecate this in favor of a native way to achieve the same thing that could launched recently.
Problem:
Goal:
Add support for ranking models in PyTorch.
New Functionality
Starting Point:
Implement base-classes of block-API in PyTorch
People: @marcromeyn
Currently the block-API is T4Rec is using a similar design to Keras to allow for modules that lazily initialize their variables. We would like to deprecate this in favor of a native way to achieve the same thing that could launched recently.
Input-blocks
People: @marcromeyn
Continuous&EmbeddingTablesTabularInputBlockOutput-blocks
People: @edknv & @marcromeyn
Blocks
Models
Examples