It would be nice to have access to the new multi-output feature exposed in `xgboost 2.0` https://xgboost.readthedocs.io/en/stable/tutorials/multioutput.html#training-with-vector-leaf