def create_state(self, index, weight):
return (mx.nd.zeros(weight.shape, weight.context, dtype=weight.dtype),  <EOL>mx.nd.zeros(weight.shape, weight.context, dtype=weight.dtype))<EOL>
Create additional optimizer state: mean, variance

        Parameters
        ----------
        weight : NDArray
        The weight data