GitHub repository: ryanhausen/jax-md
Primary language: Jupyter Notebook
Description: Differentiable, Hardware Accelerated, Molecular Dynamics