jax.lax.select ============== .. currentmodule:: jax.lax .. autofunction:: select