Puzzle Transformers