@article{jaxmdendtoenddifferentiablehardware,
  title = {JAX MD: End-to-End Differentiable, Hardware Accelerated, Molecular Dynamics in Pure Python},
  author = {Samuel S. Schoenholz and Ekin D. Cubuk},
  year = {2019},
  url = {https://openreview.net/forum?id=r1xMnCNYvB},
}