Example borrowed from: https://github.com/huggingface/kernels/tree/main/skills/cuda-kernels