stainx
stainx_cuda_torch
