stainx
stainx_cuda_cupy
stainx_cuda_torch
