"""FlashHead"""

from efficient_heads.flash_head import FlashHead
from efficient_heads.midx_head import MIDXHead
from efficient_heads.svd_softmax import SVDSoftmaxHead
from efficient_heads.vocab_pruning import VocabPruningHead
