jax_less

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

Test whether the left operand is smaller, mirroring jax.numpy.less.

Parameters:

  • x1 – Left operand.

  • x2 – Right operand.

Returns:

The elementwise boolean result of x1 < x2 for native operands, or the result supplied by a custom override.