No support for stratified split in dask_ml.model_selection.train_test_split

scikit-learn implementation of train test split (sklearn.model_selection.train_test_split) supports splitting data according to class labels (stratified split) by using the argument `stratify`. This is especially useful when datasets have high class imbalance. It would be really helpful to have this feature in dask_ml as well.