Metadata-Version: 2.1
Name: mzt
Version: 1.2.0
Summary: Package to host DeepLearning modules for pytorch ecosystem, to ease out model implementations.
Home-page: https://github.com/ShivamPR21/MZT.git
Author: Shivam Pandey
Author-email: pandeyshivam2017robotics@gmail.com
License: AGPLv3+
Keywords: DeepLearning Pytorch Modules
Classifier: Development Status :: 3 - Alpha
Classifier: Topic :: Utilities
Classifier: License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)
License-File: LICENSE

# ModuleZooTorch

Collection of neural-net modules introduced in latest research

## Installation

```shell
pip install mzt
```

## Available Modules

- Convolutional 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
