Extend the sampler from invariant measures to time-dependent laws or SDE solutions
Extend the amortized neural sampler from invariant measures to the full time-dependent law $\mu_t$ of a stochastic differential equation, or to the weak solution of the stochastic differential equation.
References
Several directions remain open. First, the supervised results rely on access to training samples from target invariant measures, which may require substantial offline simulation effort, and the amortized approach is most useful when many related SDE instances must be solved and mixing is slow. Data preparation, unsupervised training, and training without samples need further investigation. Second, the current framework targets the invariant measure, and a natural extension is to learn the full law $\mu_t$ that depends on time, or the weak solution of the SDE.