If you don't need GPU support, vigra is a great option for ND image processing.

conda install -c conda-forge vigra

python -c 'import vigra; help(vigra.analysis.watersheds)'

scikit-image ain't bad, and has more functions, but the functions vigra is faster if it has the function you're looking for.

Next on my list of ND array libraries to learn is xtensor.