Fixes a bug: passing jacobian tensors as `kwargs` in forward passes of `nnj`

MachineLearningLifeScience / stochman

Algorithms for computations on random manifolds made easier

Apache License 2.0

85 stars 11 forks source link

Closed miguelgondu closed 2 years ago

miguelgondu commented 2 years ago

Solves the bug described in #21, and adds a test for this scenario.

The solution isn't pythonic, so I'm happy to refactor the logic a little bit more.