Hi @BramVanroy I think we actually agree. What I am saying, in mathematical terms, is that the linear layers performs the linear operation
A * B
where A is a number of obs x 768 matrix and B is a 768 x number of classes matrix.
Thus the result of the matrix multiplication is number of obs  x number of classes whose row i gives the score  for each class for observation i (before actually transforming to probabilities via a softmax).
Does that make sense?