Follow-up work since the original zi2zi, now with transformer backbone.