visual-transformer
Mathematical Morphology-based self-attention module for PyTorch (CUDA) using Flash-style kernel fusion.