code: def test_mlp(getkey):
      mlp = eqx.nn.MLP(2, 2, 512, 2, key=getkey())
      mlp_noinline = eqx.internal.noinline(mlp)
      mlp_jit = eqx.filter_jit(mlp, donate="none")
      mlp_jit_noinline = eqx.filter_jit(mlp_noinline, donate="none")
      x = jr.normal(getkey(), (2,))
      o1 = mlp(x)
      o2 = mlp_jit(x)
      o3 = mlp_noinline(x)
      o4 = mlp_jit_noinline(x)
      assert tree_allclose(o1, o2)
      assert tree_allclose(o1, o3)
      assert tree_allclose(o1, o4)
  
file_path: equinox/tests/test_noinline.py
imports: import equinox
  import jax
  import jax.numpy
  import jax.random
  from .helpers import tree_allclose
label_desc: Use image node_id 2 for calling a global function with example usage: test_mlp(getkey) without return types