jat A (probably messy and naive) pure JAX implementation of a graph attention network. Copyright Copyright (c) 2022, Benjamin Kaminow Acknowledgements Project based on the Computational Molecular Science Python Cookiecutter version 1.6.