The log likelihood gradient computation of `MixtureModel` is unstable. See this [blog post](https://gregorygundersen.com/blog/2020/02/09/log-sum-exp/) for improvement.