code: def get_random_rotation_matrix():
      rotation = np.eye(4)
      rotation[1:, 1:] = scipy.stats.ortho_group.rvs(3)
      return 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 1 for calling a global function with example usage: get_random_rotation_matrix() and returns: rotation