English

GraphNeuralNetworks.jl: Deep Learning on Graphs with Julia

Machine Learning 2024-12-10 v1

Abstract

GraphNeuralNetworks.jl is an open-source framework for deep learning on graphs, written in the Julia programming language. It supports multiple GPU backends, generic sparse or dense graph representations, and offers convenient interfaces for manipulating standard, heterogeneous, and temporal graphs with attributes at the node, edge, and graph levels. The framework allows users to define custom graph convolutional layers using gather/scatter message-passing primitives or optimized fused operations. It also includes several popular layers, enabling efficient experimentation with complex deep architectures. The package is available on GitHub: \url{https://github.com/JuliaGraphs/GraphNeuralNetworks.jl}.

Keywords

Cite

@article{arxiv.2412.06354,
  title  = {GraphNeuralNetworks.jl: Deep Learning on Graphs with Julia},
  author = {Carlo Lucibello and Aurora Rossi},
  journal= {arXiv preprint arXiv:2412.06354},
  year   = {2024}
}

Comments

Submitted to JMLR OSS

R2 v1 2026-06-28T20:27:40.762Z