search

CarloLucibello / GraphNeuralNetworks.jl

Graph Neural Networks in Julia
https://carlolucibello.github.io/GraphNeuralNetworks.jl/dev/
MIT License
209 stars 47 forks source link

Lux support? #372

Open CarloLucibello opened 5 months ago

CarloLucibello commented 5 months ago

This package is built on top of Flux.jl, but we could consider also supporting Lux.jl since it is being used in some part of the ecosystem. It is not clear if supporting both can be done without a lot of code duplication.

cc @avik-pal

avik-pal commented 5 months ago

The easiest way to do that would be to provide NNlib like primitives instead of defining (l::GCNConv)(....), etc.

CarloLucibello commented 2 months ago

I'm going to factor out a DL-framework agnostic GNNlib.jl package.