Add

Adds other to input, scaled by alpha.

Abstract Signature:

Add(input: Tensor, other: Tensor, alpha: float = 1)

PyTorch

API: torch.add
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.add
Strategy: Macro 'jax.numpy.add({input}, {other} * {alpha})'

NumPy

API: numpy.add
Strategy: Macro 'numpy.add({input}, {other} * {alpha})'

Keras

API: keras.ops.add
Strategy: Macro 'keras.ops.add({input}, {other} * {alpha})'

TensorFlow

API: tf.math.add
Strategy: Macro 'tf.math.add({input}, {other} * {alpha})'

Apple MLX

API: mlx.core.add
Strategy: Macro 'mlx.core.add({input}, {other} * {alpha})'

NVIDIA SASS

API: FADD
Strategy: Direct Mapping

AMD RDNA

API: v_add_f32
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.numpy.add
Strategy: Macro 'jax.numpy.add({input}, {other} * {alpha})'

PaxML / Praxis

API: jax.numpy.add
Strategy: Macro 'jax.numpy.add({input}, {other} * {alpha})'