This is beyond my expertise and involves some words that I don’t understand, but my understanding is that quite a few people in ML (eg Michael Jordan) specifically care about things like gradient flows in the space of probability measures, and have research questions involving quite challenging differential geometry. You’d have to browse his papers to get a better understanding as I’m not qualified to expand much on this subject.