Full Implementation of DIET

Does this PyTorch implementation include every component described in the original paper?

DIET architecture
http://bl.ocks.org/koaning/raw/f40ca790612a03067caca2bde81e7aaf/

Here is the parameters defined in RASA's implementation:
https://github.com/RasaHQ/rasa/blob/main/rasa/nlu/classifiers/diet_classifier.py#L147

I don't see any of them available in this PyTorch implementation.
