code: def compute_psnr(mse):
      """Compute psnr value given mse (we assume the maximum pixel value is 1).
  
      Args:
        mse: float, mean square error of pixels.
  
      Returns:
        psnr: float, the psnr value.
      """
      return -10.0 * jnp.log(mse) / jnp.log(10.0)
  
imports: import collections
  from concurrent import futures
  import contextlib
  import functools
  import time
  from typing import List, Union
  import jax
  from jax import tree_util
  import jax.numpy
  import numpy
  from scipy import interpolate
  from scipy.spatial import transform
  import tqdm
label_desc: Use image node_id 5 for calling a global function with example usage: compute_psnr(mse) and returns: unknown