Implement ResNet in terms of the Haiku public API.
Implement ResNet in terms of the Haiku public API.

PiperOrigin-RevId: 312666961
Change-Id: I04c063994d31a912955bae37505cf762db4bf106

deepmind/dm-haiku,deepmind/dm-haiku