search

NVIDIA-Merlin / models

Merlin Models is a collection of deep learning recommender system model reference implementations
https://nvidia-merlin.github.io/models/main/index.html
Apache License 2.0
262 stars 50 forks source link

Adding MMOE & PLE #1173

Closed marcromeyn closed 1 year ago

marcromeyn commented 1 year ago

Goals :soccer:

This PR introduces mixture-of-experts + PLE/CGC. With this we should be able to write a pytorch version of the multi-task blogpost.

github-actions[bot] commented 1 year ago

Documentation preview

https://nvidia-merlin.github.io/models/review/pr-1173