Jraph - 一个用于 jax 中图形神经网络的库。
第一个是一个教育性的合作项目,其中介绍了图神经网络、图论,展示了如何使用 Jraph 解决一系列问题。请查看 这里。第二个是一个完全可用的示例,其中介绍了使用 Jraph 与 OGBG-MOLPCBA 的最佳实践,并附带了一些出色的可视化效果。请查看 这里。感谢 Lisa Wang、Nikola Jovanović 和 Ameya Daigavane。
Jraph（发音为“长颈鹿”）是一个用于在 jax 中处理图神经网络的轻量级库。它提供了图形的数据结构、用于处理图形的工具集以及一系列可分叉的图形模型。