Benchmarks of the two surrogates for the AFTERGLOWPY Gaussian jet model. We show the error distributions of the surrogate predictions against a test data set of size n_test = 7500. The different rows show the error across different passbands. The left panels show the distribution of the mean squared error as defined in Eq. (9). The right panels show the mismatch distribution across the test data set as defined in Eq. (10). The figure compares two different surrogates: one using the MLP architecture (blue) and the other a cVAE (green).