Potential OpenGL/Vulkan target?

[arunoruto]

I recently stumbled across the library Taichi, which targets more visual rendering but can also be used for scientific computation. They support the usual targets like CPUs, CUDA, and AMD GPU but also offer more generic stuff like OpenGL and Vulkan targets! This enabled me to run some example code on more exotic platforms without any issues (so long as the backend was installed correctly).
I know that supporting the right target makes it more efficient, but having such fallbacks is awesome! Also, while it has some good GPU hardware, its software support is lackluster... ROCm always has some problems, so OpenGL/Vulkan support was quite handy!
Are there any problems to have such support for JAX in the future? I am not saying you need to implement this feature, but maybe there is some interest in this from the community standpoint. I also don't know how easy it is to translate scientific workload onto these graphical libraries, so feel free to correct me :)