Pass optional named arguments to the wrapped sonnet module.
Pass optional named arguments to the wrapped sonnet module.

Named arguments can be used to change the behavior of sonnet modules. For
example, it's not uncommon to use dropout to regularize a deep neural network.
However, dropout should only be used during the training of the network, not
afterwards.

PiperOrigin-RevId: 208786599

deepmind/sonnet,deepmind/sonnet