JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling - View it on GitHub