fix: Add filterwarnings for jax.xla_computation DeprecationWarning

[matthewfeickert]

Description

Add an ignore to filterwarnings to avoid a DeprecationWarning on jax.xla_computation in jax v0.4.30+.

DeprecationWarning: jax.xla_computation is deprecated. Please use the AOT APIs.

c.f. jax-ml/jax#21923

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

* Add an ignore to filterwarnings to avoid a DeprecationWarning
  on jax.xla_computation in jax v0.4.30+.

  > DeprecationWarning: jax.xla_computation is deprecated.
  > Please use the AOT APIs.

   - c.f. https://github.com/google/jax/pull/21923

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 98.21%. Comparing base (997e5e5) to head (53d1722).
Report is 9 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #2523   +/-   ##
=======================================
  Coverage   98.21%   98.21%           
=======================================
  Files          69       69           
  Lines        4543     4543           
  Branches      804      804           
=======================================
  Hits         4462     4462           
  Misses         48       48           
  Partials       33       33           

Flag	Coverage Δ
contrib	97.79% <ø> (ø)
doctest	98.08% <ø> (ø)
unittests-3.10	96.23% <ø> (ø)
unittests-3.11	96.23% <ø> (ø)
unittests-3.12	96.23% <ø> (ø)
unittests-3.8	96.25% <ø> (ø)
unittests-3.9	96.27% <ø> (ø)

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

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.