quantax.utils.jit\_chunk\_vmap ============================== .. currentmodule:: quantax.utils .. autofunction:: jit_chunk_vmap