torchjd icon indicating copy to clipboard operation
torchjd copied to clipboard

feat(autogram): Use new engine

Open ValerianRey opened this issue 3 months ago • 1 comments

Already making the PR to keep track of progress, and to be able to update the dev-new-engine branch easily when main is updated.

  • [x] Remove explicit batched optimizations #470
  • [ ] Add and use SLT #484
  • [ ] Use no cross-terms (may be directly mergeable in main) #468
  • [ ] Implement node-based algorithm and change interface

ValerianRey avatar Oct 29 '25 15:10 ValerianRey

Codecov Report

:white_check_mark: All modified and coverable lines are covered by tests.

Files with missing lines Coverage Δ
src/torchjd/autogram/_engine.py 100.00% <100.00%> (ø)
src/torchjd/autogram/_gramian_computer.py 100.00% <100.00%> (ø)
src/torchjd/autogram/_jacobian_computer.py 100.00% <100.00%> (ø)
src/torchjd/autogram/_module_hook_manager.py 100.00% <100.00%> (ø)
:rocket: New features to boost your workflow:
  • :snowflake: Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

codecov[bot] avatar Oct 29 '25 15:10 codecov[bot]