starter: adaptive_pooling <- this works in torch and tensorflow backend, not jax.