Conv1d with some builtin handling of asymmetric or causal padding and normalization.

Wrapper around Conv2d and normalization applied to this conv to provide a uniform interface across normalization approaches.

SEANet decoder. Args: channels (int): Audio channels. dimension (int): Intermediate representation dimension. n_filters (i

SEANet encoder. Args: channels (int): Audio channels. dimension (int): Intermediate representation dimension. n_filters (i

Residual block from SEANet model. Args: dim (int): Dimension of the input/output kernel_sizes (list): List of kernel sizes for the

LSTM without worrying about the hidden state, nor the layout of the data. Expects input as convolutional layout.

ArithmeticCoder, Let us take a distribution `p` over `N` symbols, and assume we have a stream of random variables `s_t` sampled from `p`. Let

ArithmeticDecoder, see `ArithmeticCoder` for a detailed explanation. Note that this must be called with **exactly** the same parameters and seque

Convolution-friendly LayerNorm that moves channels to last dimensions before running the normalization and moves them back to original positi

STFT sub-discriminator. Args: filters (int): Number of filters in convolutions in_channels (int): Number of input channels. Defaul

Codebook with Euclidean distance. Args: dim (int): Dimension. codebook_size (int): Codebook size. kmeans_init (bool): Whet

Multi-Scale STFT (MS-STFT) discriminator. Args: filters (int): Number of filters in convolutions in_channels (int): Number of inpu

Wrapper around Conv1d and normalization applied to this conv to provide a uniform interface across normalization approaches.

Wrapper around ConvTranspose1d and normalization applied to this conv to provide a uniform interface across normalization approaches.

Residual vector quantization implementation. Follows Algorithm 1. in https://arxiv.org/pdf/2107.03312.pdf

Residual Vector Quantizer. Args: dimension (int): Dimension of the codebooks. n_q (int): Number of residual vector quantizers used

ConvTranspose1d with some builtin handling of asymmetric or causal padding and normalization.

Vector quantization implementation. Currently supports only euclidean distance. Args: dim (int): Dimension codebook_size (int)

Wrapper around ConvTranspose2d and normalization applied to this conv to provide a uniform interface across normalization approaches.