which is why we created Hidet, open-source deep learning compiler, written in Python and which can be easily extended to create all kinds of ML projects (wink, wink) with support for Nvidia GPUs. I am still looking for information on Mojo support of Nvidia GPUs, de-facto compute standard for modern AI/ML.

https://docs.hidet.org/stable/gallery/developer-guides/hidet...