Don't run stateful mixed precision example in doctest.
Don't run stateful mixed precision example in doctest.

PiperOrigin-RevId: 289848542
Change-Id: I66743dd05dccdac8a8b063b73995601a6c430652

deepmind/sonnet,deepmind/sonnet