jit pow_n and sqrt_n