grad_TauMatrix_LL_constrained is even faster now