PyTorch
API: —
Strategy: Macro 'torch.square({x} - {y})'
JAX (Core)
API: —
Strategy: Macro 'jnp.square({x} - {y})'
NumPy
API: —
Strategy: Macro 'numpy.square({x} - {y})'
Keras
API: —
Strategy: Macro 'keras.ops.square({x} - {y})'
TensorFlow
API: tf.math.squared_difference
Strategy: Direct Mapping
Apple MLX
API: —
Strategy: Macro 'mlx.core.square({x} - {y})'
Flax NNX
API: —
Strategy: Macro 'jnp.square({x} - {y})'
PaxML / Praxis
API: —
Strategy: Macro 'jnp.square({x} - {y})'