# Split separates structure (GraphDef) from data (State)
graphdef, state = nnx.split(model)

# Merge reconstructs a live module from structure and data
reconstructed_model = nnx.merge(graphdef, state)

        __
        __