Currently `thermox.sample` will break if `D` is all zeros, but it might be nice to support exact simulation of the linear ODE. Note this doesn't apply to `thermox.log_prob` as the linear ODE is not probabilistic.