fully_shard() has a check that seems a bit overzealous, what about modules that inherit from ModuleList or ModuleDict for the list/dict accessors that provide their own forward? If the goals it to ...