* add zero init context * add more flags for zero init context fix bug of repeated converting param to ShardedParamV2 * polish code