jax_divmod¶
- probly.representation.jax_functions.jax_divmod(x1: JaxArrayLike, x2: JaxArrayLike, /) tuple[jax.Array, jax.Array][source]¶
- probly.representation.jax_functions.jax_divmod(x1: JaxOperand, x2: JaxOperand, /) object
Compute quotients and remainders, mirroring
jax.numpy.divmod.- Parameters:
x1 – Dividend.
x2 – Divisor.
- Returns:
A tuple of floor-division quotients and remainders for native operands, or the result supplied by a custom override.