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