Triton is pretty hot right now you can check it out.

https://openai.com/blog/triton/

https://triton-lang.org/master/index.html