jax

d69fe1f8 - Merge pull request #24873 from jax-ml:no-gen-linear-util