-
-
Notifications
You must be signed in to change notification settings - Fork 46
Open
Description
For the past few weeks, I've been working on issues related to NEP-18 support for Dask (and ultimately, Dask-GLM as well) to allow CuPy to be used as a backend. I've made some progress, which I'll soon share with more details.
Among the issues I have now is the usage of the following Dask-GLM algorithms with CuPy:
newton
: requires numpy.linalg.lstsqadmm
andlbfgs
: require scipy.optimize.fmin_l_bfgs_b
Both of these functions are not implemented in CuPy. So I'm wondering if anybody knows whether any of these functions have some open and reliable CUDA implementation that could be integrated into CuPy or if we could implement them somewhat easily on top of already supported CuPy functionality?
Metadata
Metadata
Assignees
Labels
No labels