xlstm_jax.models.xlstm_clean.utils

Classes

UpProjConfigMixin

Module Contents

class xlstm_jax.models.xlstm_clean.utils.UpProjConfigMixin

proj_factor: float = None

round_proj_up_dim_up: bool = True

round_proj_up_to_multiple_of: int = 64

_proj_up_dim: int = None

_set_proj_up_dim(embedding_dim)

Parameters:

embedding_dim (int)

Return type:

None