Check out our paper on amortized solutions of model-constrained Bayesian inverse problems using surrogate-driven measure transport.