Never got how mamba models work in multiple dimensions and non-causally.