jaxchem An experimental repo to work on some Jax models for chemistry Setup Development Docker $ git clone https://github.com/deepchem/jaxchem.git $ cd jaxchem $ docker build . -t jaxchem