I think Rapids AI's cuML tried to go into this direction (essentially scikit-learn on the GPU): https://docs.rapids.ai/api/cuml/stable/api.html#logistic-regress

I think Rapids AI's cuML tried to go into this direction (essentially scikit-learn on the GPU): https://docs.rapids.ai/api/cuml/stable/api.html#logistic-regression https://docs.rapids.ai/api/cuml/stable/api.html#logistic-reg.... For some reason it never took really off though. Btw., going on a tangent, you might like Hummingbird (https://github.com/microsoft/hummingbird https://github.com/microsoft/hummingbird). It allows you trained scikit-learn tree-based models to PyTorch. I watched the SciPy talk last year, and it's a super smart & elegant idea.