code: def _check_ntks(ntks):
      # Check that implementations match
      for ntk1 in ntks:
          for ntk2 in ntks:
              diff = tf.reduce_max(tf.abs(ntk1 - ntk2))
              print(f"NTK implementation diff {diff}.")
              assert diff < 1e-4, diff
  
      print("All NTK implementations match.")
  
imports: from absl import app
  import neural_tangents
  import tensorflow
label_desc: Use image node_id 2 for calling a global function with example usage: _check_ntks(ntks) without return types