How to use sigmoid-neuron/flash-attention-1-triton with Kernels:
# !pip install kernels from kernels import get_kernel kernel = get_kernel("sigmoid-neuron/flash-attention-1-triton")
drbh