Support for Sparse Matrices on GPU (CUSPARSE)

Specific Domains GPU
, ,
1

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

2 Likes
3

There used to be GitHub - JuliaAttic/CUSPARSE.jl: Julia interface to NVIDIA's CUSPARSE library, 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.