chore(lib): make plotting utilities accept JAX arrays

[jeertmans]

No description provided.

[github-actions]

[LanguageTool] _{reported by reviewdog 🐶}
If a new sentence starts here, add a space and start with an uppercase letter. (LC_AFTER_PERIOD[1])
Suggestions: Vertices, vertices
Rule: https://community.languagetool.org/rule/show/LC_AFTER_PERIOD?lang=en-US&subId=1
Category: CASING

[github-actions]

[LanguageTool] _{reported by reviewdog 🐶}
If a new sentence starts here, add a space and start with an uppercase letter. (LC_AFTER_PERIOD[1])
Suggestions: Triangles, triangles
Rule: https://community.languagetool.org/rule/show/LC_AFTER_PERIOD?lang=en-US&subId=1
Category: CASING

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 80.10%. Comparing base (ef9013e) to head (b9daffb).
Report is 90 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #146      +/-   ##
==========================================
+ Coverage   79.34%   80.10%   +0.76%     
==========================================
  Files          25       25              
  Lines        1753     1820      +67     
==========================================
+ Hits         1391     1458      +67     
  Misses        362      362              

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

[codspeed-hq]

CodSpeed Performance Report

Merging #146 will not alter performance

_{Comparing avoid-asarray (b9daffb) with main (ef9013e)}

Summary

✅ 15 untouched benchmarks