grad_TauMatrix_LL_constrained is faster again