def cumprod(x, axis = None): return scan(prims.multiply, x, axis = axis)
def cumprod(x, axis=None): return scan(prims.multiply, x, axis=axis)
def cumsum(x, axis = None): return scan(prims.add, x, axis = axis)
def cumsum(x, axis=None): return scan(prims.add, x, axis=axis)