/jraph

A Graph Neural Network Library in Jax

Primary LanguagePythonApache License 2.0Apache-2.0

Watchers