code: def get_random_higgs_photons():
      higgs = 125.18
      boost = get_random_boost_matrix()
      rotation = get_random_rotation_matrix()
      photon1 = (
          boost @ rotation @ np.array([higgs / 2, higgs / 2, 0, 0])
      )
      photon2 = (
          boost @ rotation @ np.array([higgs / 2, -higgs / 2, 0, 0])
      )
      return photon1, photon2
  
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 3 for calling a global function with example usage: get_random_higgs_photons() and returns: photon1,  photon2