HesScale: Scalable Computation of Hessian Diagonals