code: def convert_bn_layer(keras_layer, objax_layer):
      """Converts variables of batch normalization layer from Keras to Objax."""
      shape = objax_layer.gamma.value.shape
      objax_layer.gamma = objax.TrainVar(
          jn.array(keras_layer.__dict__["gamma"].numpy()).reshape(shape)
      )
      objax_layer.beta = objax.TrainVar(
          jn.array(keras_layer.__dict__["beta"].numpy()).reshape(shape)
      )
      objax_layer.running_mean = objax.StateVar(
          jn.array(keras_layer.__dict__["moving_mean"].numpy()).reshape(
              shape
          )
      )
      objax_layer.running_var = objax.StateVar(
          jn.array(
              keras_layer.__dict__["moving_variance"].numpy()
          ).reshape(shape)
      )
  
imports: import functools
  from typing import Callable, Sequence, Union, Optional
  from jax import numpy
  import objax
  from objax.constants import ConvPadding
  from objax.nn import Conv2D
  from objax.typing import JaxArray, ConvPaddingInt
label_desc: Use image node_id 2 for calling a global function with example usage: convert_bn_layer(keras_layer, objax_layer) without return types