fix: ensure bounds vmaps as best as possible, more equinox errors, types for __eq__, _BoundsD/I, and _PositionD/I#250
fix: ensure bounds vmaps as best as possible, more equinox errors, types for __eq__, _BoundsD/I, and _PositionD/I#250beckermr wants to merge 82 commits into
__eq__, _BoundsD/I, and _PositionD/I#250Conversation
Merging this PR will improve performance by 45.35%
|
__eq____eq__, _BoundsD/I
|
OK @ismael-mendoza this one is ready for review. I hope this is the last major bit of work on these classes for a long time! |
__eq__, _BoundsD/I__eq__, _BoundsD/I, and _PositionD/I
This PR fixes a few other things with the bounds classes, images and typing.
BoundsD/Ivmaps as best as possible__eq__methods return JAX boolean values except when no values from the class are ever traced (e.g., two staticBoundsIclasses, interpolants)_BoundsD/I_PositionD/I__eq__APIcloses #245
closes #237