Weld as a Foundation for ML in Rust

[Tiberio]

Hello, thank you for your work. I'm currently looking for a suitable solution for building Machine Learning foundation libraries in Rust.

I'm evaluating how Weld could help in this space by generating optimised code for CPU and GPU by leveraging it's IR. However, I couldn't find much information about how to use Weld to generate GPU code.

Is this something in progress? Where can I find more info about it?