xlstm_jax.models.xlstm_pytorch.blocks.mlstm.backend

Submodules

• xlstm_jax.models.xlstm_pytorch.blocks.mlstm.backend.config
• xlstm_jax.models.xlstm_pytorch.blocks.mlstm.backend.config_utils
• xlstm_jax.models.xlstm_pytorch.blocks.mlstm.backend.fwbw
• xlstm_jax.models.xlstm_pytorch.blocks.mlstm.backend.layer_factory
• xlstm_jax.models.xlstm_pytorch.blocks.mlstm.backend.simple
• xlstm_jax.models.xlstm_pytorch.blocks.mlstm.backend.tl_utils
• xlstm_jax.models.xlstm_pytorch.blocks.mlstm.backend.triton_chunk

Attributes

_mlstm_backend_registry
create_mlstm_backend

Classes

mLSTMBackendNameAndKwargs

Package Contents

xlstm_jax.models.xlstm_pytorch.blocks.mlstm.backend._mlstm_backend_registry

class xlstm_jax.models.xlstm_pytorch.blocks.mlstm.backend.mLSTMBackendNameAndKwargs

Bases: config_utils.NameAndKwargs

_registry: dict[str, type]

name: str = ''

kwargs: dict[str, Any] | None

get_config_class_for_kwargs()

Return type:

type

get_class_for_name()

Return type:

type

create_name_instance()

Return type:

Any

xlstm_jax.models.xlstm_pytorch.blocks.mlstm.backend.create_mlstm_backend