The non-abelian Hodge locus