Mixture Density Networks (MDN) implemented in PyTorch - View it on GitHub