GitHub repository: vishalbelsare/jraph
Primary language: Python
Description: A Graph Neural Network Library in Jax