xlstm_jax.models.xlstm_clean.blocks.mlstm.block

Classes

mLSTMBlockConfig

Functions

mLSTMBlock(config, *args, **kwargs)

Module Contents

class xlstm_jax.models.xlstm_clean.blocks.mlstm.block.mLSTMBlockConfig

mlstm: xlstm_jax.models.xlstm_clean.blocks.mlstm.layer.mLSTMLayerConfig

_num_blocks: int | None = None

_block_idx: int | None = None

xlstm_jax.models.xlstm_clean.blocks.mlstm.block.mLSTMBlock(config, *args, **kwargs)

Parameters:

config (mLSTMBlockConfig)

Return type:

flax.linen.Module