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