code: def test_poolbackprop():
      def max_pool_mean(x):
          max_pool = eqx.nn.MaxPool3d(
              (2, 2, 2), (1, 1, 1), ((0, 1), (0, 1), (0, 1))
          )
          return jnp.mean(max_pool(x))
  
      x = jnp.arange(64.0, dtype=jnp.float32).reshape(1, 4, 4, 4)
      grad_fn = jax.value_and_grad(max_pool_mean)
  
      grad_fn(x)
  
imports: import warnings
  from typing import Union
  import equinox
  import jax
  import jax.nn
  import jax.numpy
  import jax.random
  import pytest
label_desc: Use image node_id 45 for calling a global function with example usage: test_poolbackprop() and returns: jnp