Good question, I think that’s why Jax has an extra compilation step where the graph derived from the program is optimized. I don’t see why that can’t happen at runtime, though i can see implementing the optimization potentially being a lot more work than just letting the compiler handle it.