code/flash-linear-attention/fla/ops/sse/__init__.py

from .index import prepare_sample_relpos_global_index_flat
from .mask import softmax_and_mask

__all__ = [
    "prepare_sample_relpos_global_index_flat",
    "softmax_and_mask",
]