How to use replicate/metal-flash-sdpa with Kernels:
# !pip install kernels from kernels import get_kernel kernel = get_kernel("replicate/metal-flash-sdpa")