Hi all, I created a MultiTaskLearner that don’t need to concat any thing.
MultiTaskLearner
You migh be interested in it.