Skip to content

[FEA] Move solvers from cuml to raft #374

Open
@cjnolet

Description

@cjnolet

There are several solvers in cuML which can benefit algorithm development across the larger RAPIDS ecosystem by living in RAFT. For example, there is interest in exposing some of the solvers from RAFT to the cupy.optimize package.

Current solvers in cuml which we have discussed moving over to RAFT:

  1. L-BFGS (also to extend w/ trust region constraints for L-BFGS-B)
  2. OWL-QN
  3. Coordinate descent
  4. SGD
  5. LARS
  6. SMO
  • Any solvers which will be exposed through the cupy.optimize package will need to accept a generalized objective function (eg functor/lambda or constructed from some set of predefined building blocks like the semirings/distance API) and will need to accept constraints.

There have also been discussions of wanting the CG, Powell, and potentially Nelder-Meade simplex methods, though these would be targeted as future features after the above solvers have been moved over.

Metadata

Metadata

Assignees

Labels

EPICThese might span releasesfeature requestNew feature or request

Type

No type

Projects

Status

In Progress

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions