speechbrain.nnet

Package containing the different neural networks layers

speechbrain.nnet.CNN	Library implementing convolutional neural networks.
speechbrain.nnet.RNN	Library implementing recurrent neural networks.
speechbrain.nnet.activations	Library implementing activation functions.
speechbrain.nnet.attention	Library implementing attention modules.
speechbrain.nnet.autoencoders	Autoencoder implementation.
speechbrain.nnet.containers	Library for implementing cascade (sequences) of different neural modules.
speechbrain.nnet.diffusion	An implementation of Denoising Diffusion
speechbrain.nnet.dropout	Library implementing dropout.
speechbrain.nnet.embedding	Library implementing embedding.
speechbrain.nnet.hypermixing	This module mixes information from different tokens via HyperMixing.
speechbrain.nnet.linear	Library implementing linear transformation.
speechbrain.nnet.losses	Losses for training neural networks.
speechbrain.nnet.normalization	Library implementing normalization.
speechbrain.nnet.pooling	Library implementing pooling.
speechbrain.nnet.quantisers	Gumbel Softmax implementation with multiple groups possible.
speechbrain.nnet.schedulers	Schedulers for updating hyperparameters (such as learning rate).
speechbrain.nnet.unet	A UNet model implementation for use with diffusion models
speechbrain.nnet.utils	Assorted reusable neural network modules.

speechbrain.nnet.complex_networks	Package containing complex neural networks
speechbrain.nnet.loss	Package containing specific losses (transducer, stoi ...)
speechbrain.nnet.quaternion_networks	Package containing quaternion neural networks
speechbrain.nnet.transducer	Package containing transducer neural networks

Summary

Reference