jax_power

probly.representation.jax_functions.jax_power(x1: JaxArrayLike, x2: JaxArrayLike, /) → jax.Array[source]
probly.representation.jax_functions.jax_power(x1: JaxOperand, x2: JaxOperand, /) → object

Raise operands to powers elementwise, mirroring jax.numpy.power.

Parameters:

  • x1 – Bases.

  • x2 – Exponents.

Returns:

x1 ** x2 elementwise, or the result supplied by a custom override.