code: def test_jax_leaks():
      with jax.checking_leaks():
          test_compose_continuous()
          test_compose_ongrid()
          test_compose_gradient()
  
file_path: jaxdf/tests/test_functions.py
imports: import jax
  import numpy
  from jax import jit
  from jax import numpy
  from jaxdf import *
label_desc: Use image node_id 10 for calling a global function with example usage: test_jax_leaks() without return types