Support for Sparse Matrices on GPU (CUSPARSE)

Do any of the ML libraries (Flux, KNet) support sparse matrices on the GPU?

2 Likes

There used to be https://github.com/JuliaGPU/CUSPARSE.jl, but that hasn’t seen maintenance in a while. Making that work well with CuArrays would allow Flux to do sparse matrices on the GPU.