jax.lax.add#

jax.lax.add(x, y)[source]#

Elementwise addition: \(x + y\).

This function lowers directly to the stablehlo.add operation.

Parameters:
  • x (ArrayLike) – Input arrays. Must have matching numerical dtypes. If neither is a scalar, x and y must have the same number of dimensions and be broadcast compatible.

  • y (ArrayLike) – Input arrays. Must have matching numerical dtypes. If neither is a scalar, x and y must have the same number of dimensions and be broadcast compatible.

Returns:

An array of the same dtype as x and y containing the sum of each pair of broadcasted entries.

Return type:

Array

See also

  • jax.numpy.add(): NumPy-style addition supporting inputs with mixed dtypes and ranks.