How to use JingzeShi/flash-sparse-attention with Kernels:
# !pip install kernels from kernels import get_kernel kernel = get_kernel("JingzeShi/flash-sparse-attention")