xlstm_jax.models.xlstm_clean.blocks.mlstm.backend.config

Classes

mLSTMBackend

Module Contents

class xlstm_jax.models.xlstm_clean.blocks.mlstm.backend.config.mLSTMBackend

Bases: flax.linen.Module

config: Any