jax.lax.rem =========== .. currentmodule:: jax.lax .. autofunction:: rem