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