jax.lax.complex =============== .. currentmodule:: jax.lax .. autofunction:: complex