JAX MD: A Framework for Differentiable Physics

Open in new window