fix: ensure bounds vmaps as best as possible, more equinox errors, types for __eq__, _BoundsD/I, and _PositionD/I

fix: ensure bounds vmaps as best as possible, more equinox errors, types for __eq__, _BoundsD/I, and _PositionD/I #901