Duplicated from kernels-community/sage_attention
How to use replicate/sage_attention with Kernels:
# !pip install kernels from kernels import get_kernel kernel = get_kernel("replicate/sage_attention")
