jax_add

probly.representation.jax_functions.jax_add(x1: JaxArrayLike, x2: JaxArrayLike, /) → jax.Array

probly.representation.jax_functions.jax_add(x1: JaxOperand, x2: JaxOperand, /) → object

Add two operands elementwise, mirroring jax.numpy.add.

Parameters:

• x1 – Left operand.

• x2 – Right operand.

Returns:

The elementwise sum, or the result supplied by a custom override.