frontend: (stablehlo) add ir generation from jaxpr

[shreya-um]

This will let us create an xDSL module from a function jitted with JAX.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 89.90%. Comparing base (4423038) to head (89525f2).
Report is 45 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #3038      +/-   ##
==========================================
+ Coverage   89.86%   89.90%   +0.03%     
==========================================
  Files         415      419       +4     
  Lines       52078    53006     +928     
  Branches     8048     8219     +171     
==========================================
+ Hits        46799    47653     +854     
- Misses       3984     4025      +41     
- Partials     1295     1328      +33     

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

[superlopuh]

This looks great! Feel free to mark the PR as "ready for review", and remove the [WIP] in the PR title

[superlopuh]

let's just assert this is true for this PR, and change it in the next one?

[superlopuh]

This is great! feel free to merge once you've made the suggested changes