A brain atlas for allenai/tmax-27b, the largest member of the tmax hybrid SSM/Mamba/transformer family. This model required custom kernel work to fit into the atlas pipeline, and the result is the deepest, most redundant, and most surgically forgiving atlas in the family.
Model: allenai/tmax-27b
Corpus: 8,965 diverse prompts
Layers probed: all 64
Attention layers: 3, 7, 11, 15, 19, 23, 27, 31, 35, 39, 43, 47, 51, 55, 59… See the full description on the dataset page:
https://huggingface.co/datasets/juiceb0xc0de/tmax-27b-atlas.