xlstm_jax.models.xlstm_parallel.blocks.mlstm.backend.config_utils#

Classes#

NameAndKwargs

Sub-model configuration.

Module Contents#

class xlstm_jax.models.xlstm_parallel.blocks.mlstm.backend.config_utils.NameAndKwargs#

Bases: xlstm_jax.models.configs.SubModelConfig

Sub-model configuration.

This class is currently a quick fix to allow for post-init style model configs, like the xlstm-clean we ported from the original xlstm codebase. Once the config system is more mature, we should remove this and all becomes a subclass of ModelConfig.

name: str = ''#

kwargs: dict[str, Any] | None#

_registry: dict[str, type] | None#

get_config_class_for_kwargs()#

Return type:: type

get_class_for_name()#

Return type:: type

create_name_instance()#

Return type:: Any

to_dict()#

Converts the config to a dictionary.

Helpful for saving to disk or logging.

Return type:: dict