jax

7e7c9de8 - Merge pull request #29966 from mattjj:direct-linearize-mutable-array