PyRank

Pytorch Modules Python Packages

Python packages with the GitHub topic pytorch-modules. Sorted by monthly downloads, with stars and forks.
vhrabar
serron

Mathematical Morphology module for PyTorch (with CUDA support), providing differentiable operators and learnable network layers.

2K 0 0
vhrabar
morphottention

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

442 0 0
v-garzon
torchresidual

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.

45 1 0