pass test for loading / writing state with multi gpu

still unclear if backprop works or not, need to test cwc-backwards