Add test for nested mixed_precision scopes.
Add test for nested mixed_precision scopes.

PiperOrigin-RevId: 274754476
Change-Id: I0f34464666a4dea7ee2e5757a8d9452f0e575a1b

deepmind/sonnet,deepmind/sonnet