Second-Order Differential Machine Learning using Jax - Kichler, Neil (2023)