Update CI test requirements.txt

[andrijapau]

Context:

Follow up to ##1084.

After merging to master, it was found that arm-gpu wheel building tests were failing (see here).

Description of the Change:
Update CIs to use requirements-tests.txt instead of requirements-dev.txt.
Also install jax for CI tests.

Benefits:
CI (arm+cuda) is 💚
no longer require installing unnecessary packages.

Possible Drawbacks:
Jax version is explicitly pinned in the CI test

[sc-85859]
[sc-87026]

[github-actions]

Hello. You may have forgotten to update the changelog!
Please edit .github/CHANGELOG.md with:

• A one-to-two sentence description of the change. You may include a small working example for new features.
• A link back to this PR.
• Your name (or GitHub username) in the contributors section.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 99.11%. Comparing base (e34523f) to head (7aaf6c0).
Report is 1 commits behind head on master.

Additional details and impacted files

@@            Coverage Diff             @@
##           master    #1105      +/-   ##
==========================================
+ Coverage   97.97%   99.11%   +1.13%     
==========================================
  Files         238      148      -90     
  Lines       41252    22979   -18273     
==========================================
- Hits        40416    22775   -17641     
+ Misses        836      204     -632     

☔ 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.

[AmintorDusko]

Looks good! Thanks!