Add ReaxFF potential

.github/workflows/build.yml

@@ -44,6 +44,7 @@ jobs:
         pip install pytest-xdist
         pip install pytest-cov
         pip install netCDF4
+        pip install ase
     - name: Set Swap Space
       uses: pierotofy/set-swap-space@master
       with:

jax_md/__init__.py

@@ -25,7 +25,6 @@
 from jax_md import util
 from jax_md import io
 from jax_md import rigid_body
-
 try:
   # Attempt to load colab_tools if IPython is installed.
   from jax_md import colab_tools

jax_md/reaxff/__init__.py

@@ -0,0 +1,4 @@
+from . import reaxff_energy
+from . import reaxff_forcefield
+from . import reaxff_interactions
+from . import reaxff_helper