JAX: Efficient Calculation of the Hessian Matrix | Hacker News Reader