The MAMBA design transformer having a language modeling head on major (linear layer with weights tied towards the enter which describes how all The inner states are related since they represent the fundamental dynamics https://k2spiceshop.com/product/liquid-k2-on-paper-online/