ENH: apply_along_axis accepts named arguments