GitHub repository: IsaacBreen/jraph
Description: A Graph Neural Network Library in Jax