Neural Network Modules - nnlib.modules

The modules package provides out-of-the-box NN modules, implemented in PyTorch.

Embedding

Linear

Softmax