This is some basic implementation of MDN as starting point for future projects.
Credit: https://engineering.taboola.com/predicting-probability-distributions/
saArbabi/MDN_initial-implementation
Predicting Probability Distributions Using Neural Networks
Jupyter NotebookMIT