reshaping -> enfold/defold
enfold in the slicing distributor relies on the fact, that the global axes correspond to the local array axes, as it operates on local data from get_local_data().
In general this is not true for generic distribution strategies.