Review for NeurIPS paper: JAX MD: A Framework for Differentiable Physics