code: def residual_block(
      scope: Scope, x: Array, conv, norm, act, features: int
  ):
      residual = x
      x = scope.child(conv, "conv_1")(x, features, (3, 3))
      x = scope.child(norm, "bn_1")(x)
      x = act(x)
      x = scope.child(conv, "conv_2")(x, features, (3, 3))
      x = scope.child(norm, "bn_2")(x)
      return act(residual + x)
  
file_path: flax/tests/core/design/core_big_resnets_test.py
imports: from functools import partial
  import jax
  import numpy
  from absl.testing import absltest
  from jax import numpy
  from jax import random
  from flax.core import Array, Scope, init, lift, nn, unfreeze
label_desc: Use image node_id 1 for calling a global function with example usage: residual_block(scope, x, conv, norm, act, features) and returns: act