NNlib.jl icon indicating copy to clipboard operation
NNlib.jl copied to clipboard

spconv ... sparse convolution

Open dorn-gerhard opened this issue 1 year ago • 1 comments

Motivation and description

For many applications (graph neural networks, 3d point clouds, ...) sparse convolution would be a cool feature

Possible Implementation

A guide to implementation is given here (with some references): https://towardsdatascience.com/how-does-sparse-convolution-work-3257a0a8fd1

dorn-gerhard avatar Jun 06 '23 15:06 dorn-gerhard

I doubt anyone is going to step up to write custom CUDA kernels for this like in that post unless they also need the feature, so it's in solid "PRs welcome" status for now. You may want to see if anything in GeometricFlux, GraphNeuralNetworks.jl or Flux3D helps with your use case. Those three should cover GNNs and point clouds between them.

ToucheSir avatar Jun 06 '23 15:06 ToucheSir