PyTorch
API: torch.nn.MaxPool2d
Strategy: Direct Mapping
Keras
API: keras.layers.MaxPool2D
Strategy: Direct Mapping
TensorFlow
API: keras.layers.MaxPool2D
Strategy: Direct Mapping
Apple MLX
API: mlx.nn.layers.pooling.MaxPool2d
Strategy: Direct Mapping
Flax NNX
API: flax.nnx.max_pool
Strategy: Direct Mapping