from nfn.common.data import ArraySpec, NetworkSpec, WeightSpaceFeatures
from nfn.common.data import state_dict_to_tensors, params_to_state_dicts, params_to_func_params
from nfn.common.data import network_spec_from_wsfeat