Example: ```python import pyfixest as pf data = pf.get_data() # no error pf.feols("Y ~ X1 | f1 + f2", data=data, solver = "jax", vcov = "iid") # no error pf.feols("Y ~ X1 | f1 + f2", data=data, solver = "jax", vcov = "hetero") # error pf.feols("Y ~ X1 | f1 + f2", data=data, solver = "jax", vcov = {"CRV1":"f1"}) # File "pyfixest\estimation\vcov_utils.py", line 131: # def _crv1_meat_loop( # <source elided> # ) -> np.ndarray: # k = scores.shape[1] ``` I suppose that there is some incompatibility between the numba code in the `_crv1_meat_loop` and jax?