Disable JAX acceleration by default

[mikemhenry]

Checklist

• All new code is appropriately documented (user-facing code must have complete docstrings).
• Added a news entry, or the changes are not user-facing.
• Ran pre-commit by making a comment with pre-commit.ci autofix before requesting review.

Manual Tests: these are slow so don't need to be run every commit, only before merging and when relevant changes are made (generally at reviewer-discretion).

• GPU integration tests
• example notebook testing
• packaging tests: run this for any large feature PRs or PRs that add test data.

Developers certificate of origin

• I certify that this contribution is covered by the MIT License here and the Developer Certificate of Origin at https://developercertificate.org/.

[mikemhenry]

TODO: This needs a news item!

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 93.16%. Comparing base (78d0f90) to head (f290ebf).
⚠️ Report is 1 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1694      +/-   ##
==========================================
- Coverage   95.47%   93.16%   -2.32%     
==========================================
  Files         187      187              
  Lines       16244    16244              
==========================================
- Hits        15509    15133     -376     
- Misses        735     1111     +376     

Flag	Coverage Δ
fast-tests	93.16% <100.00%> (?)
slow-tests	?

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.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[mikemhenry]

We decided to add a warning that links to a document page that describes this behavior.
If we have time, we will investigate a solution that re-sets the envar to what it was before we changed it/set it.

[mikemhenry]

Actually I forgot that when you set an envar within python it only exists for the python process and doesn't stay around:
$ echo "$PYMBAR_DISABLE_JAX"; python -c "import os; os.environ.setdefault('PYMBAR_DISABLE_JAX', 'TRUE')"; echo "$PYMBAR_DISABLE_JAX"

this prints 2 blank lines for me, if it stayed around, it would print a blank line and a line that says TRUE

[IAlibay]

TIL Python is a shell with its own local env variables. Thanks @mikemhenry !

[mikemhenry]

to save you a click (also TIL our theme makes github URLs look nice! I just used a raw URL)

[mikemhenry]

pre-commit.ci autofix

[atravitz]

agree with adding an example error output, otherwise LGTM!

[mikemhenry]

Good idea, will make searching the docs more useful -- let me know if this looks good @IAlibay @atravitz

[mikemhenry]

actually I need to add a news entry, I will edit the JAX warning one

[github-actions]

No API break detected ✅