/graph_NNs_jax

Exploratory project into Graph Neural Networks (GNNs) and various Jax resources.

Primary LanguagePython

Description:

Exploratory work into (convolutional) GNNs for node classification, as well as flax and jraph.