Jraph-jax中的图神经网络库。 | Jraph（发音为“长颈鹿”）是一个轻量级的库，用于在jax中使用图神经网络。 它提供了图的数据结构，一组使用图的工具以及可插拔图神经网络模型的“动物园”。 安装 可以使用以下命令直接从github安装Jraph： pip install git+git://github.com/deepmind/jraph.git 概述 Jraph旨在提供用于在jax中处理图形的实用程序，但并未规定编写或开发图形神经网络的方法。 graph.py提供了一个轻