jax
971d34f8 - Simplify the handling of closed-over constants in Jaxprs.

Commit
254 days ago
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.
Author
Committer
Parents
Loading