JAX, M.D. A Framework for Differentiable Physics

Open in new window