There are many papers on ML for query planners. You can search for "Learned Query Optimization". Some use ML just for the cardinality estimation.