Repository navigation

perf(autojac): Add optimize_gramian_computation to jac_to_grad - #525

Merged

ValerianRey merged 38 commits into

mainfrom

optimize_jac_to_grad

Feb 23, 2026

Merged

perf(autojac): Add optimize_gramian_computation to jac_to_grad#525
ValerianRey merged 38 commits into
mainfrom
optimize_jac_to_grad

Commits on Jan 23, 2026
- WIP: add gramian-based jac_to_grad
  ValerianRey
  committed
  ae6be7d
- Update changelog
  ValerianRey
  committed
  8bdf512
- Use deque to free memory asap
  ValerianRey
  committed
  aaf2544
- Merge branch 'main' into optimize_jac_to_grad
  ValerianRey
  authored
  64b06ad
Commits on Jan 28, 2026
- Merge branch 'main' into optimize_jac_to_grad
  ValerianRey
  authored
  745f707
- Merge branch 'main' into optimize_jac_to_grad
  ValerianRey
  authored
  5eb77f9
- Use gramian_weighting in jac_to_grad
  ValerianRey
  committed
  8f65caa
- Merge branch 'main' into optimize_jac_to_grad
  ValerianRey
  authored
  6fe15a4
Commits on Jan 29, 2026
- Merge branch 'main' into optimize_jac_to_grad
  ValerianRey
  authored
  d5cb5c2
- Only optimize when no forward hooks
  ValerianRey
  committed
  f986950
- Make _gramian_based take aggregator instead of weighting
  ValerianRey
  committed
  4cf5cbb
- Add _can_skip_jacobian_combination helper function
  ValerianRey
  committed
  add549c
- Add test_can_skip_jacobian_combination
  ValerianRey
  committed
  453971a
- Optimize compute_gramian for when contracted_dims=-1
  ValerianRey
  committed
  9d4c41c
Commits on Jan 30, 2026
- Merge branch 'main' into optimize_jac_to_grad
  ValerianRey
  authored
  48cd70b
- Use TypeGuard in _can_skip_jacobian_combination
  
  ValerianRey
  and
  PierreQuinton
  authored
  8f2660d
- [pre-commit.ci] auto fixes from pre-commit.com hooks
  pre-commit-ci[bot]
  committed
  fc9bbcf
Commits on Feb 1, 2026
- Merge branch 'main' into optimize_jac_to_grad
  ValerianRey
  authored
  3f9a6d1
Commits on Feb 4, 2026
- Merge branch 'main' into optimize_jac_to_grad
  ValerianRey
  authored
  9d9cbf0
- [pre-commit.ci] auto fixes from pre-commit.com hooks
  pre-commit-ci[bot]
  committed
  b5ca226
Commits on Feb 5, 2026
- Merge branch 'main' into optimize_jac_to_grad
  ValerianRey
  authored
  0baa914
Commits on Feb 13, 2026
- Merge branch 'main' into optimize_jac_to_grad
  ValerianRey
  authored
  2ed1d7c
- [pre-commit.ci] auto fixes from pre-commit.com hooks
  pre-commit-ci[bot]
  committed
  86be778
- Merge branch 'main' into optimize_jac_to_grad
  ValerianRey
  authored
  4ace19e
- Add ruff if-else squeezing
  ValerianRey
  committed
  2a84bef
Commits on Feb 23, 2026
- Merge branch 'main' into optimize_jac_to_grad
  ValerianRey
  authored
  4b6209c
- [pre-commit.ci] auto fixes from pre-commit.com hooks
  pre-commit-ci[bot]
  committed
  1b1c660
- Many fixes of problems coming from the merge
  ValerianRey
  committed
  b714253
- Fix _can_skip_jacobian_combination
  ValerianRey
  committed
  2bb8ab1
- Make check_consistent_first_dimension work with Deque
  ValerianRey
  committed
  63c9dde
- Improve test_can_skip_jacobian_combination
  ValerianRey
  committed
  0f85811
- Add optimize_gramian_computation param and add error when not compatible
  ValerianRey
  committed
  9d55215
- Fix overloads (partly) and add missing code coverage
  ValerianRey
  committed
  456510b
- Fix overloads
  ValerianRey
  committed
  55c69d1
- Fix docstring
  ValerianRey
  committed
  8a401a3
- fixup what @ValerianRey did wrong
  PierreQuinton
  committed
  b4bf7c4
- Improve error message
  ValerianRey
  committed
  24a991a
- Improve docstring
  ValerianRey
  committed
  2ea44a4

Commits

Commits on Jan 23, 2026

Commits on Jan 28, 2026

Commits on Jan 29, 2026

Commits on Jan 30, 2026

Commits on Feb 1, 2026

Commits on Feb 4, 2026

Commits on Feb 5, 2026

Commits on Feb 13, 2026

Commits on Feb 23, 2026