really speeding up inference by computing quadratic form with diagonal plus low rank using eingsum