RiemannianSSMs See research/test_model_benchmark.jl for a full example of the optimised inference implementation.