* add save/load state dict for ColoDDP * add unit test * refactor unit test folder * polish unit test * rename unit test