Simplify the handling of closed-over constants in Jaxprs.
In this version, we make more non-scalar arrays literable, so that
the constants get embedded directly into the equattions.
This corresponds to Attempt 2 from #29679.
This is just an experiment.