Skip to content

Support differentiation through LM solver#53

Draft
chungmin99 wants to merge 9 commits intobrentyi:mainfrom
chungmin99:feat/differentiable
Draft

Support differentiation through LM solver#53
chungmin99 wants to merge 9 commits intobrentyi:mainfrom
chungmin99:feat/differentiable

Conversation

@chungmin99
Copy link
Copy Markdown
Contributor

This PR adds a differentiable solver API, and demonstrates it in an end-to-end IMU fusion calibration example (Full-LM vs Deep-LM).

  • Introduces solve_differentiable() with implicit/adjoint gradients through the inner LM solve,
  • New differentiable solve path using custom_vjp and one adjoint linear solve (conjugate_gradient or dense_cholesky).
  • Proper tangent/cotangent handling for manifold variables in backprop through the solve.
  • A full sensor-calibration example with outer-loop bias learning via Optax and trajectory/bias comparisons.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant