Here are
3 public repositories
matching this topic...
Flexible residual connections for PyTorch with a clean API. Build complex skip connections using Record/Apply modules. 5 operations (add, concat, gated, highway, multiply), learnable alphas, automatic projection. Thread-safe for DataParallel/DDP. Fully typed and tested.
Updated
Feb 18, 2026
Python
Mathematical Morphology-based self-attention module for PyTorch (CUDA) using Flash-style kernel fusion.
Updated
Jul 17, 2026
Cuda
Mathematical Morphology module for PyTorch (with CUDA support), providing differentiable operators and learnable network layers.
Updated
Jul 25, 2026
Cuda
Improve this page
Add a description, image, and links to the
pytorch-modules
topic page so that developers can more easily learn about it.
Curate this topic
Add this topic to your repo
To associate your repository with the
pytorch-modules
topic, visit your repo's landing page and select "manage topics."
Learn more
You can’t perform that action at this time.