Comparison of fwd pass of Bi-Mamba2 v. Naive Flipping and Running Mamba2 kernel twice.
Comparison of fwd pass of Bi-Mamba2 v. causal Mamba2.
Comparison of bwd pass of Bi-Mamba2 v. Naive Flipping and Running Mamba2 kernel twice.
...
Comparison of bwd pass of Bi-Mamba2 v. causal Mamba2.
...
Comparison of bwd pass of Bi-Mamba2 v. Naive Flipping v. Causal Mamba2.
Coming Soon
Coming Soon