stochastic_control: optimize bellman equations by looping over distribution only once