main
py 7 lines 276 Bytes
Raw
1 # SPDX-License-Identifier: Apache-2.0
2 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3 """Fused Triton kernels for late-interaction (MaxSim) scoring."""
4
5 from .flash_maxsim_rerank import flash_maxsim_rerank_direct
6
7 __all__ = ["flash_maxsim_rerank_direct"]