jax_remainder¶
- probly.representation.jax_functions.jax_remainder(x1: JaxArrayLike, x2: JaxArrayLike, /) jax.Array[source]¶
- probly.representation.jax_functions.jax_remainder(x1: JaxOperand, x2: JaxOperand, /) object
Compute floor-division remainders, mirroring
jax.numpy.remainder.- Parameters:
x1 – Dividend.
x2 – Divisor, whose sign determines the remainder’s sign.
- Returns:
The elementwise remainder, or the result supplied by a custom override.