JAX time (no JIT): 0.0094 seconds

        __
        __