/jax-ghnet

Graph Highway Networks in JAX

Primary LanguagePython

Stargazers