For compute shaders, you can already do this with CubeCL: https://github.com/tracel-ai/cubecl https://github.com/tracel-ai/cubecl You write kernels in a Rust D

For compute shaders, you can already do this with CubeCL: https://github.com/tracel-ai/cubecl https://github.com/tracel-ai/cubecl You write kernels in a Rust DSL using #[cube], it supports WebGPU through WGSL and Vulkan through SPIR-V, along with CUDA, AMD via ROCm, and Metal. (disclosure: I am a contributor)