For elastic distortion applied to 3D images, the time spent on elastic deformation (which is typically implemented in NumPy, e.g. as in https://gist.github.com/nasimrahaman/8ed04be1088e228c21d51291f47dd1e6 48) is almost 1min/image. I wonder if there is already an implementation in PyTorch using cuda that speeds things up.