shredx.modules#

Modules

cnn

mlp

Multi-layer perceptron (MLP) encoders and decoders for sequence modeling.

moe_mixin

Mixture of Experts mixin module for SINDy layer forecasting.

positional_encoding

Sinusoidal positional encoding for transformer models.

rnn

Recurrent neural network encoders for sequence modeling.

sindy_layer

SINDy Layer module.

sindy_loss_mixin

SINDy Loss mixin module for neural network regularization.

transformer

Transformer encoders for sequence modeling.