jax.jit ======= .. currentmodule:: jax .. autofunction:: jit