[mlpack] GSoC ’22 Idea Discussion: Add CUDA Support For mlpack

Ryan Curtin ryan at ratml.org
Tue Apr 12 11:13:21 EDT 2022


On Mon, Apr 11, 2022 at 11:55:00PM +0800, Zhuojin Liu wrote:
> The worry I have is whether mlpack is ready for adding GPU supports now.
> Though it's on the mlpack vision list, but it's not on the GSoC idea list, plus
> that the bandicoot library is not release-ready now, so I'm not sure it's time
> to add GPU support to mlpack this year.

Indeed, bandicoot is still in an early stage.  At least personally I
think it would be reasonable to propose a project to complete some
functionality there, if you can find an interested mentor.  In essence
bandicoot becomes ready when we have 'enough' functionality implemented
(where 'enough' is defined as the set of functions we need to, e.g.,
implement some simple neural networks in mlpack using coot::mat as the
matrix type).

-- 
Ryan Curtin    | "Like, with jetpacks?"
ryan at ratml.org |   - Scott


More information about the mlpack mailing list