BibTeX
@INPROCEEDINGS{
         Schoenholz2020JMA,
       author = "Schoenholz, Samuel and Cubuk, Ekin Dogus",
       booktitle = "Advances in Neural Information Processing Systems",
       editor = "H. Larochelle and M. Ranzato and R. Hadsell and M. F. Balcan and H. Lin",
       pages = "11428--11441",
       publisher = "Curran Associates, Inc.",
       title = "{JAX MD}: A Framework for Differentiable Physics",
       url =
         "https://proceedings.neurips.cc/paper/2020/file/83d3d4b6c9579515e1679aca8cbc8033-Paper.pdf",
       volume = "33",
       year = "2020",
       ad_area = "Molecular Dynamics",
       ad_tools = "JAX"
}