I think the sum_axis is a good naming scheme (see map_axis, fold_axis). - [x] Add new names `sum_axis`, `mean_axis` - [x] Rename `scalar_sum` → `sum`.