jraph:Jax中的图神经网络库

Jraph-jax中的图神经网络库。 | Jraph（发音为“长颈鹿”）是一个轻量级的库，用于在jax中使用图神经网络。 它提供了图的数据结构，一组使用图的工具以及可插拔图神经网络模型的“动物园”。 安装 可以使用以下命令直接从github安装Jraph： pip install git+git://github.com/deepmind/jraph.git 概述 Jraph旨在提供用于在jax中处理图形的实用程序，但并未规定编写或开发图形神经网络的方法。 graph.py提供了一个轻量级的数据结构GraphsTuple ，用于处理图形。 utils.py提供了在jax中使用GraphsTuples 。 批处理GraphsTuples数据集的实用工具。 通过填充和遮罩支持可变形状图形的jit编译的实用程序。 用于定义输入分区损失的实用程序。 models.py提供了不同