code: def get_random_background_photon():
      boost = get_random_boost_matrix()
      rotation = get_random_rotation_matrix()
      energy = np.random.uniform(20, 120)
      return boost @ rotation @ np.array([energy, energy, 0, 0])
  
file_path: jraph/jraph/examples/higgs_detection.py
imports: import collections
  import logging
  import random
  from absl import app
  import haiku
  import jax
  import jax.numpy
  import jraph
  import numpy
  import optax
  import scipy.stats
label_desc: Use image node_id 4 for calling a global function with example usage: get_random_background_photon() and returns: unknown