The NNX API reference does not seem to have entries for pooling operations like `nnx.avg_pool` or `nnx.max_pool`.