Method transform

python/tests/test_compile.py:1242–1243

(x: State)

            b: mx.array

        def transform(x: State):
            return State(x.a + 10, x.b * 10)

        def transform_tuple(t):
            return (t[0] + 10, t[1] * 10)