BoxOSQP with vmap very slow

[ZedongPeng]

I am using batched BoxOSQP. However, it's very slow. It's even slower than CvxpyQP with a for loop.
Does anyone have any experience in using and accelerating batched BoxOSQP?

osqp_batch_jit_solve = jax.vmap(jax.jit(osqp_run))

def osqp_run():
    ...
    osqp = BoxOSQP(tol=eps, verbose=0, maxiter=4000, jit=True, implicit_diff=False, unroll=False)
    sol = osqp.run(None, params_obj=(Q, new_c_vecter), params_eq=(A, b), params_ineq=(G, h)).params