There's a comparison of JAX with PyTorch for Hessian calculation here! https://www.assemblyai.com/blog/why-you-should-or-shouldnt-be-using-jax-in-2022/#hessian

There's a comparison of JAX with PyTorch for Hessian calculation here! https://www.assemblyai.com/blog/why-you-should-or-shouldnt-be-using-jax-in-2022/#hessians https://www.assemblyai.com/blog/why-you-should-or-shouldnt-b... Would definitely be interested in an article like that if you decide to write it