models

models copied to clipboard

### Description Create an example notebook on how to use Multi-Task Learning (MTL) with Merlin Models. **Additional context** - Models currently provides support to MMoE and PLE blocks, which creates...

gabrielspmoreira

[RMP] Performant large embedding table support

2

Model Parallel Support - [ ] ~Evaluation of HugeCTR, TorchRec, Distributed Embeddings, TFRA, PersiaML for inclusion in Merlin~ - [ ] Distributed embedding table support in merlin-models (SOK Plugin, Distributed...

EvenOldridge

Example of serving Implicit in Merlin Systems

3

viswa-nvidia

[Task] Example for Debugging Merlin Models

1

### Description I want to propose an example, how to debug merlin model architectures and functionality. Content could be (not limited to it): - viewing model.layers, explaining the structure -...

bschifferer

[TASK] Support shared embeddings across the towers

2

# 🚀 Feature request We should be able to jointly encode multiple columns (e.g., purchased_id, item_id) and let them share the same embedding tables during model training, both for ranking...

rnyak

[BUG] The negative sampling for ranking gives AUC 0.00 value when we add `sampling` layer to the Model class

5

### Bug description The negative sampling for ranking gives AUC 0.00 value and binary classification acc 0.999 when we add `sampling` layer to the `mm.Model()` class. ### Steps/Code to reproduce...

rnyak

Part of #669 Relates to https://github.com/NVIDIA-Merlin/Merlin/issues/545 ### Background Based on the discussions around having a common API across the backend framework implementations. We would like to make this concrete with...

oliverholworthy

[FEA] Save input and output schema when `.save` methods are called on models

3

Relates to: https://github.com/NVIDIA-Merlin/Merlin/issues/545 # 🚀 Feature request Provide a consistent `.save` interface for all models. This `.save` method should save the model artifact along with the schema to a directory...

oliverholworthy

InputBlockV2 - Extend test coverage

2

Currently there are dozens of tests using `InputBlock`. This task is about creating copies of those tests and replacing `InputBlock` by `InputBlockV2`, which over time will be the de-facto solution...

gabrielspmoreira

Add a protocol for expected interface of a model and save input schema to metadata directory

21

Fixes #676 and #677 ### Goals :soccer: - Improve consistency and clarity of models API. - adding protocol to document expected methods on a Model. - Enable use of Merlin...

oliverholworthy