torchjd
torchjd copied to clipboard
feat(autogram): Use new engine
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
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.