Cache subjaxprs when lowering to MLIR functions.
Cache subjaxprs when lowering to MLIR functions.

This will only actually dedup the subjaxpr when
jax_experimental_subjaxpr_lowering_cache is enabled.

google/jax,google/jax,google/jax,google/jax