gauss-newton-method
numerical optimization in pytorch
Composable Second-Order Optimization for JAX and Optax.