@ev-br I just gave the ndimage tests a whirl on GPU. Looks like you covered CuPy and JAX well, but there are a lot of failures with PyTorch. We need to figure out how to skip on GPU while still testing CuPy. The crude way to do this is to just add skips for torch specifically , which may be the best option for now.

https://gist.github.com/lucascolley/62e223fa893204ffc51da057457a641b