add 1st pytorch example

[radekosmulski]

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) is more informative to the reader (this is the language used in the paper, otherwise it is a bit confusing what does output refer to -- is it the output of the model? of mlp hidden layers? embedding_dim is more informative)

[github-actions[bot]]

Documentation preview

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

[review-notebook-app[bot]]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB