xlstm_jax.models.xlstm_parallel.blocks.mlstm.layer_v1

Classes

mLSTMLayerV1

mLSTM layer with a Transformer-style projection up and down.

Module Contents

class xlstm_jax.models.xlstm_parallel.blocks.mlstm.layer_v1.mLSTMLayerV1

Bases: flax.linen.Module

mLSTM layer with a Transformer-style projection up and down.

config: xlstm_jax.models.xlstm_parallel.blocks.mlstm.layer.mLSTMLayerConfig