PyTorch
API: torch.nn.AvgPool1d
Strategy: Direct Mapping
JAX (Core)
API: —
Strategy: Custom / Partial
Keras
API: keras.layers.AvgPool1D
Strategy: Direct Mapping
TensorFlow
API: tf.keras.layers.AvgPool1D
Strategy: Direct Mapping
Apple MLX
API: —
Strategy: Custom / Partial
Flax NNX
API: flax.nnx.AvgPool
Strategy: Direct Mapping
PaxML / Praxis
API: —
Strategy: Custom / Partial