JuliaGraphs/GraphNeuralNetworks.jl

What it solves

Julia 언어를 사용하여 그래프 구조 데이터에 대해 딥러닝을 수행하기 위한 라이브러리 세트를 제공합니다. 사용자는 Flux.jl 및 Lux.jl과 같은 인기 있는 딥러닝 프레임워크를 사용하여 그래프 신경망(GNNs)을 구축하고 학습시킬 수 있습니다.

How it works

이 프로젝트는 여러 전문화된 패키지로 구성되어 있습니다:

• GraphNeuralNetworks.jl 및 GNNLux.jl은 각각 Flux 및 Lux 사용자를 위한 주요 사용자 인터페이스 역할을 합니다.
• GNNGraphs.jl은 기반이 되는 그래프 데이터 구조와 헬퍼 함수를 처리합니다.
• GNNlib.jl은 gather/scatter 메커니즘 또는 희소 행렬 곱셈을 사용하여 핵심 메시지 전달 프레임워크를実装합니다.

Who it’s for

그래프 컨볼루션 레이어를 사용하여 노드, 엣지 또는 그래프 수준에서 머신러닝 작업을 구현하고자 하는 Julia 사용자(연구자 및 개발자).

Highlights

• 일반적인 그래프 컨볼루션 레이어를 지원하며 사용자 정의 레이어 정의를 지원합니다.
• 배치 그래프(batched graphs), 이종 그래프(heterogeneous graphs) 및 동적 그래프(dynamical graphs)에 대한 계산 능력을 제공합니다.
• CUDA 및 AMDGPU를 통한 하드웨어 가속을 지원합니다.
• Graphs.jl 생태계와 통합됩니다.