It would be nice to use `SymbolicIndexingInterface.jl` to enable indexing such as ```julia sol[rn.A^2 * rn.B] ``` to extract the moment `E[A^2 B]`.