code: def mse_loss(
      targets: jnp.ndarray, predictions: jnp.ndarray
  ) -> jnp.ndarray:
      """Implementation of the mean squared error loss."""
      return jnp.power((targets - predictions), 2)
  
imports: from flax import linen
  import jax.numpy
label_desc: Use image node_id 2 for calling a global function with example usage: mse_loss(targets, predictions) and returns: jnp