jax

8ff63961 - Merge pull request #4686 from google:lazy-simplification-again