shredx.modules.mlp#

Multi-layer perceptron (MLP) encoders and decoders for sequence modeling. Also implements MOE-MLP encoder.

Classes

MLPDecoder(in_dim, out_dim, n_layers, dropout)

Multi-Layer Perceptron (MLP) decoder.

MLPEncoder(input_size, hidden_size, ...[, ...])

Multi-layer perceptron (MLP) encoder.

MOEMLPEncoder(input_size, hidden_size, ...)

Multi-Layer Perceptron (MLP) with SINDy layer forecasting.

SINDyLossMLPEncoder(input_size, hidden_size, ...)

MLP encoder with SINDy loss regularization.