How to set weights to a sonnet nets module?

Question

I am using the DeepMind Sonnet Library, https://sonnet.readthedocs.io/en/latest/

After defining a simple MLP, I run it with a mock datapoint in order to set the input shape.

import tensorflow as tf
import sonnet as snt

mlp = snt.nets.MLP([32, 32, 1])
mlp(np.array([[1.0, 1.0, 1.0]]))  # MLP has structure 3 -> 32 -> 32 -> 1

I am now looking for a way to set my predefined weights for this network. Something like mlp.set_weights() for instance. Is there a way to do that?