In-Context Operator Learning on the Space of Probability Measures

Frank Cole, Dixi Wang, Yineng Chen +2

We introduce \emph{in-context operator learning on probability measure spaces} for optimal transport (OT). The goal is to learn a single solution operator that maps a pair of distr…