models

models copied to clipboard

Add HybridQA dataset

1

Depends on #1206.

edknv

Add SentencePieceTokenizer and LlamaTokenizer

1

edknv

Adding LoRA as the first peft-method

1

Fixes # (issue) ### Goals :soccer: ### Implementation Details :construction: ### Testing Details :mag:

marcromeyn

add 1st pytorch example

2

This adds the first pytorch example 🥳 Additionally, I propose in `DLRMModel` we rename `dim` to `embedding_dim`. This aligns us with the tf api and (which is probably more important)...

radekosmulski

Moving tf examples to tf-folder

1

Fixes # (issue) ### Goals :soccer: ### Implementation Details :construction: ### Testing Details :mag:

marcromeyn

MM Pytorch API: TabularTransform for input tabular sequence

1

Fixes # (issue) ### Goals :soccer: - Add support for padding, transforming, and masking sequential inputs data in MM Pytorch backend - The implemented transform classes should: - Support multiple...

sararb

Introduce distributed embeddings

5

Part of https://github.com/NVIDIA-Merlin/Merlin/issues/733. ### Goals :soccer: There is a package called [distributed-embeddings](https://github.com/NVIDIA-Merlin/distributed-embeddings), a library for building large embedding based (e.g. recommender) models in Tensorflow. It's an alternative approach to SOK....

edknv

### Description Priority of examples for new PyTorch API 1. [ ] Getting Started 2. [ ] Explore Different Ranking Models 3. [ ] Retrieval Models 4. [ ] ecommerce-next-item-prediction...

bschifferer

add a topK operation on top of softmax layer.

EvenOldridge

### Description **Additional context** coming soon...

rnyak