Certainly! We do so :)
https://blog.tensorflow.org/2019/04/mlir-new-intermediate-re...
In general we can afford to treat the problem as a search problem since our kernels are small enough. Here there is a lot of stuff that ML aided stochastic search is amazing at compared to normal compilers and even in some cases human hand optimized assembly (!).