Hi, first off, kudos! This module is simply awesome.
Is there any interest in adding GPU support for optimizations in the near future?
If so, this module (cuML) might be useful.
I didn't dig too deep into it, but their first key concept is "Where possible, match the scikit-learn API¶", so it might be easier than it looks.