search

patrick-kidger / lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
Apache License 2.0
365 stars 24 forks source link

a vs _a #99

Closed lockwo closed 5 months ago

lockwo commented 5 months ago

In the docs, mv uses _a but as matrix uses a. Are these meaningfully different parameters or could they be made the same?

Screenshot 2024-06-04 at 9 17 28 AM
patrick-kidger commented 5 months ago

The underscore means that it is not checked. This reflects the fact that it's in a PyTree, and different nodes in the PyTree need not have the exact same sizes.