• Model merger.
• Loss function weighted sum vs independently.
• Printing of training evolution.
• Save model (I think code already available in ml4chem could be used for that).
• Set convergence stop mechanism.
• Parallelization.
• Add more compatibility with different available components.
• Check for consistency of the class (independent vs dependent loss).

That class would help to merge models like this: