Valérian Rey

TODO: - [ ] Add a special type of error for that, like UnsupportedOnSparseError. We could catch that + NotImplementedError, but not catch ValueError for instance. - [ ] Prioritize...

feat(autogram): Use new engine

1

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...

refactor(autogram): Remove vmap rule generation

3

Not sure what I'm doing here, but as far as I understand, the generated vmap rule for JacobianAccumulator is never used. I think (correct me if I'm wrong) that it...

feat(autogram): Do not consider cross-terms

2

* Add and use JacobianBasedGramianComputerWithoutCrossTerms * Make ModelAlsoUsingSubmoduleParamsDirectly and InterModuleParamReuse xpass * Fix missing type hint * Test against no cross-terms