Auto-regressive transformer