Defining the spatial organization of tissues and organs like the brain from large datasets is a major challenge. Here, authors introduce CellTransformer, an AI tool that defines spatial domains in the mouse brain based on spatial transcriptomics, a technology that measures which genes are active in different parts of tissue.