jax

bfa581b7 - Merge pull request #30351 from jax-ml:direct-linearize-changelog-notes