jax

7025b7d1 - Merge pull request #26363 from mattjj:mutable-array-error-tweak