'Create additional optimizer state: mean, variance
Parameters
weight : NDArray
The weight data'

def create_state(self, index, weight):

    return (mx.nd.zeros(weight.shape, weight.context, dtype=weight.dtype), mx.nd.zeros(weight.shape, weight.context, dtype=weight.dtype))
