Make mu more general

[ksunden]

Perhaps go to using an OrderedDict for mu, rather than the list which aligns with the states of rho

I was originally under the impression that that would work, but realize now that it does not. We need a way to track the label of mu with the value. Direct indexing is nice for portability to C implementation for CUDA. (The labels won't be tracked)