generic/LogSoftMax speedups