search

patrick-kidger / equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Apache License 2.0
1.86k stars 129 forks source link

Graph Neural Networks #702

Open stergiosba opened 1 month ago

stergiosba commented 1 month ago

Greetings Patrick,

I was wondering, what are your thoughts when it comes to Graph Neural Networks? Would a GNNlayer be implemented as a module for Equinox or an advanced example for the documentation?

Sorry if this has been brought up before, I couldn't find a similar issue/question.

patrick-kidger commented 1 month ago

I probably wouldn't add it to the core library, but I'd be happy to have that as one of our examples!