the beginnings of a more robust version of the transformer