Fixed a number of errors in log_sum_exp penalties and gradients