So the DistArray is saved in the h5py as a ndarray? Does that mean we don't have to worry about saving/loading with a different number of engines?

Could you add access to `fromndarray`'s kwargs: `dist` and `grid_shape`.