MCPcopy Create free account
hub / github.com/ml-explore/mlx / transform

Method transform

python/tests/test_vmap.py:781–782  ·  view source on GitHub ↗
(x: State)

Source from the content-addressed store, hash-verified

779 b: mx.array
780
781 def transform(x: State):
782 return State(x.a + 10, x.b * 10)
783
784 def transform_tuple(t):
785 return (t[0] + 10, t[1] * 10)

Callers

nothing calls this directly

Calls 1

StateClass · 0.70

Tested by

no test coverage detected