fix: Use jax.errors API as jax._src removed from public namespace

[matthewfeickert]

Description

Resolves #2033

This only affects the tests as jax.errors.TracerArrayConversionError works in the JAX v0.2.10 API, so nothing needs to be altered elsewhere.

Checklist Before Requesting Reviewer

• Tests are passing
• "WIP" removed from the title of the pull request
• Selected an Assignee for the PR to be responsible for the log summary

Before Merging

For the PR Assignees:

• Summarize commit messages into a comprehensive review of the PR

* Use jax.errors API over jax._src.errors as jax._src is removed from the
  jax public namespace in jax v0.3.18.
   - c.f. https://github.com/google/jax/releases/tag/jax-v0.3.18

[codecov]

Codecov Report

Base: 98.30% // Head: 98.30% // No change to project coverage 👍

Coverage data is based on head (6164930) compared to base (2d791e8).
Patch has no changes to coverable lines.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #2034   +/-   ##
=======================================
  Coverage   98.30%   98.30%           
=======================================
  Files          69       69           
  Lines        4530     4530           
  Branches      738      738           
=======================================
  Hits         4453     4453           
  Misses         45       45           
  Partials       32       32           

Flag	Coverage Δ
contrib	27.52% <ø> (ø)
doctest	61.14% <ø> (ø)
unittests-3.10	96.31% <ø> (ø)
unittests-3.7	96.28% <ø> (ø)
unittests-3.8	96.33% <ø> (ø)
unittests-3.9	96.35% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.

☔ View full report at Codecov.
📢 Do you have feedback about the report comment? Let us know in this issue.

[matthewfeickert]

As this is just updating an API only used in the tests and that passes on the minimum supported dependencies workflow, I'm going to approve and merge this myself. As always, PRs approved by a single core dev can be reverted as needed by the rest of the dev team.