TorchMorph: CUDA-Accelerated Morphology Library for PyTorch

Kai Zhao released TorchMorph, a PyTorch extension providing GPU-accelerated morphological and distance transform operators up to 8D with a SciPy-compatible API, optimized for GPU training loops via fused CUDA kernels.

2026-08-26 ~ 2026-08-26 · 2 related posts