jax

071da567 - Merge pull request #22187 from selamw1:array_equal_doc