code: def get_random_boost_matrix():
      eta = np.random.uniform(-1, 1)
      boost = np.eye(4)
      boost[:2, :2] = np.array(
          [[np.cosh(eta), -np.sinh(eta)], [-np.sinh(eta), np.cosh(eta)]]
      )
      rotation = get_random_rotation_matrix()
      return rotation.T @ boost @ rotation
  
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 2 for calling a global function with example usage: get_random_boost_matrix() and returns: unknown