Package to host DeepLearning modules for pytorch ecosystem, to ease out model implementations.
Project description
ModuleZooTorch
Collection of neural-net modules including both general blocks, and those introduced in latest research
The library is still under development, if you have some cool NN blocks in mind that deserves to be implemented in MZT please don't hesitate to open an issue with feature request, I'll try my best to implement those ASAP.
Installation
pip install mzt
Available Modules
- Convolution Blocks:
- ConvNormActivation1d
- ConvNormActivation2d
- ConvInvertedBlock1d
- ConvInvertedBlock2d
- Residual Blocks:
- ConvResidualBlock1d
- ConvResidualBlock2d
- ConvBottleNeckResidualBlock1d
- ConvBottleNeckResidualBlock2d
- ConvInvertedResidualBlock1d
- ConvInvertedResidualBlock2d
- Attention Blocks:
- SelfAttention1d
- SelfAttention2d
- MultiHeadSelfAttention1d
- MultiHeadSelfAttention2d
- MultiHeadAttention1d
- MultiHeadAttention2d
- MultiHeadAttentionLinear
- Graphical/Geometric Blocks:
- GraphConv2d
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
mzt-2.0.0.tar.gz
(22.5 kB
view hashes)
Built Distribution
Close
Hashes for mzt-2.0.0-cp312-cp312-manylinux_2_35_x86_64.whl
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 720f4167c178bd2caddebc3184e36fd9dfa44be611b537734c6eea913887a6a6 |
|
| MD5 | 6c32c546603d3d38bf52f48195aadaa1 |
|
| BLAKE2b-256 | 4174bd3b1ae3cdc94deb255fa85b02b60ad945e90239c6884601d02abad32411 |
