Custom Vulkan / GLSL Extension tutorial

Hi,

I have seen the “Custom C++ and CUDA Extensions” tutorial here:
https://pytorch.org/tutorials/advanced/cpp_extension.html

However, it doesn’t cover Vulkan / GLSL yet. Is expanding the tutorial for Vulkan / GLSL planned?

In the meantime, would anyone have pointers on how to best do extend PyTorch with a GLSL shader similar to those already in the main repo (pytorch/aten/src/ATen/native/vulkan/glsl at master · pytorch/pytorch · GitHub)?

Many thanks,
Tom