I like the h2o.ai tool for ml. It is java but it is familiar and does a decent job.
Here is info about stratified splitting in general: