Skip to content

Commit

Permalink
rm hessian - still unstable
Browse files Browse the repository at this point in the history
  • Loading branch information
Charles MOUSSA committed Mar 9, 2025
1 parent c6f28e7 commit a5c9c55
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions tests/test_differentiation.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,10 +118,3 @@ def shots_dm(x):
grad_shots = d_shots(x)

assert jnp.allclose(grad_backprop, grad_shots, atol=SHOTS_ATOL)

dd_exact = jax.hessian(lambda x: exact(x).sum())
dd_gpsr = jax.hessian(lambda x: exact_gpsr(x).sum())

gradgrad_backprop = dd_exact(x)
gradgrad_gpsr = dd_gpsr(x)
assert jnp.allclose(gradgrad_backprop, gradgrad_gpsr, atol=0.01)

0 comments on commit a5c9c55

Please # to comment.