Added _logsum_axis to _hmmc to reduce temporary memory usage