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 (CUDA), providing differentiable operators and learnable network layers.
Updated
Jul 18, 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.