1: \begin{abstract}

2: We present the implementation of a score-matching neural network that represents a data-driven prior for non-parametric galaxy morphologies.

3: The gradients of this prior can be included in the optimization routine of the recently developed multi-band modeling framework \scarlettwo, a redesign of the \scarlet method currently employed as deblender in the pipelines of the HyperSuprimeCam survey and the Rubin Observatory. The addition of the prior avoids the requirement of non-differentiable constraints, which can lead to convergence failures we discovered in \scarlet. We present the architecture and training details of our score-matching neural network and show with simulated Rubin-like observations that \scarlettwo outperforms \scarlet in accuracy of total flux and morphology estimates, while maintaining excellent performance for colors. We also demonstrate significant improvements in the robustness to inaccurate initializations. \scarlettwo is written in \texttt{python}, extendend by \texttt{jax} and \texttt{equinox}, and is fully GPU compatible. The implementation and data package of the score model are publicly available at \url{https://github.com/pmelchior/scarlet2}.

4: \end{abstract}

5: