How to use Deep-ML/flash-attention-in-cuda-from-scratch with Kernels:
# !pip install kernels from kernels import get_kernel kernel = get_kernel("Deep-ML/flash-attention-in-cuda-from-scratch")
Deep-ML commited on 