Custom XGBoost Loss function w/ Zygote. Julia Computing blog post

Thanks!
In my code above I can automatically create a list of all matching models w/

m_match = models(matching(X, y), x -> x.prediction_type == :deterministic);

Is there an equivalent in your package?