[pallas:sc] Extracted `lower_jaxpr_into_module`
This allows lowering multiple kernels into the same module. For example, we
might have separate kernels for the scalar and vector subcores.
The rest of the plumbing in Pallas SC is not yet ready for this, but we will
be building up from here :)
PiperOrigin-RevId: 868123498
