jax.lax.eq ========== .. currentmodule:: jax.lax .. autofunction:: eq